n2cholas/awesome-jax
A curated list of awesome JAX libraries, models, tutorials, papers, and videos for machine learning research.
Not currently ranked — collecting fresh signals.
star history
This repository aggregates resources for JAX, Google’s machine learning framework that combines automatic differentiation with XLA compilation. The list includes neural network libraries like Flax, Haiku, and Objax; pre-built models and projects; tutorials; research papers; and video resources. It serves as a community-maintained reference for researchers and developers working with JAX for deep learning.
Frequently asked
- What is n2cholas/awesome-jax?
- A curated list of awesome JAX libraries, models, tutorials, papers, and videos for machine learning research.
- Is awesome-jax open source?
- Yes — n2cholas/awesome-jax is open source, released under the CC0-1.0 license.
- How popular is awesome-jax?
- n2cholas/awesome-jax has 2.1k stars on GitHub.
- Where can I find awesome-jax?
- n2cholas/awesome-jax is on GitHub at https://github.com/n2cholas/awesome-jax.