Penzai

https://github.com/google-deepmind/penzai

Penzai is a JAX library for writing models as legible, functional pytree data structures with tools...

Tokens:81,046
Snippets:908
Trust Score:8.7
License:Apache-2.0
Update:1 month ago
Tokens:
Raw