awesome-jax

Why we included this project

If you work in JAX and keep losing track of which library does what, this curated index is a genuinely useful starting point. It gathers the ecosystem's neural-network frameworks (Flax, Haiku, Equinox), optimization and probabilistic-programming tools, and research projects into one navigable list, so you can quickly find the right building block for a new experiment. The maintainers also separate battle-tested libraries from newer, less-proven ones, which is a practical signal when you are deciding what to build on. Beyond code, the list points to tutorials, papers, videos, and books, making it a reasonable first stop for anyone ramping up on JAX or surveying what the ecosystem offers. It is a reference resource rather than software you deploy, but for discovery and orientation it saves real time.