← all repositories

n2cholas/awesome-jax

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

2.1k stars LearningML Frameworks
awesome-jax
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.