This is a collection of JAX utilities that are shared across several of my research projects, and is not intended as a public facing API. That said, feel free to make use of any that are useful, with all of the usual caveats.
-
Notifications
You must be signed in to change notification settings - Fork 0
jonny-so/jaxutil
About
Common JAX utilities
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published