n2cholas/awesome-jax
A curated list of awesome JAX libraries, models, tutorials, papers, and videos for machine learning research.

Velocity · 7d
+1.1
★ / day
Trend
→steady
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.