/jax-models

Unofficial JAX implementations of deep learning research papers

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers