JAX
A curated list of awesome libraries, projects, tutorials, and resources for the JAX machine learning ecosystem.
What is JAX?
Awesome JAX is a curated list of resources for the JAX machine learning ecosystem. It compiles libraries, models, tutorials, papers, and community links to help developers and researchers discover tools and learn how to use JAX effectively. The project addresses the challenge of navigating the fast-growing JAX landscape by providing a centralized, community-maintained directory.
Machine learning researchers, data scientists, and developers who are using or exploring JAX for high-performance computing, automatic differentiation, and accelerator-based (GPU/TPU) model training.
It saves time by aggregating and categorizing the best JAX resources in one place, is community-driven to ensure freshness, and provides a structured overview of the entire ecosystem, from core libraries to niche research tools.
Overview
JAX - A curated list of resources https://github.com/google/jax
Use Cases
Best For
- Finding JAX libraries for specific tasks like neural networks, optimization, or probabilistic programming
Related Projects
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
