Training & fine-tuning
JAX
Treats differentiation, compilation and vectorization as composable transformations of numerical functions.
Overview
Research summary
JAX transforms Python and NumPy-style numerical functions for machine learning and scientific computing. Its APIs support automatic differentiation, just-in-time compilation and vectorization, and can be composed to build customized numerical programs. XLA compilation targets accelerators such as GPUs and TPUs. Higher-level libraries, including Flax and Optax, build neural-network and optimization workflows on these foundations. JAX itself focuses on array computation and program transformations rather than prescribing a particular training framework or model architecture.
Repository summary
- Stars
- 36,370
- Open issues
- Unavailable
- Last push
- 2026-10-02
- Commits, 90 days
- Unavailable
- Repository activity
- Not scored
- Version
- Unavailable
Recorded catalogue figures. View repository data and provenance →
Classification
Pricing & services
Paid services unknown
Whether the provider offers paid products or services has not been established.
Licence scope
Implementation
Recorded implementation details and interfaces for JAX.
Implementation details
Python library and supporting tools
- Languages
- Python
- Repository type
- source
Recorded interfaces and capabilities
Licence scope
Repository
Repository snapshots, release information and recorded maintenance signals.
Repository snapshot
- Stars
- 36,370
- Open issues
- Unavailable
- Last push
- 2026-10-02
- Commits, 90 days
- Unavailable
- Repository activity
- Not scored
- Archived
- Not recorded
Repository activity is a snapshot, not a quality or popularity ranking. It combines recent-push freshness (50%), 90-day commits (30%) and issue pressure (20%).
Maintenance and provenance
- Catalogue snapshot
- 2026-10-06
- Stars source
- Recorded fallback
- Last push source
- Recorded fallback
Documentation
Recorded references and research provenance for this entry.
Recorded sources 5
- https://docs.jax.dev Project page · Documentation
- https://github.com/jax-ml/jax Linked repository · Research reference
- https://api.github.com/repos/jax-ml/jax Research reference
- https://github.com/jax-ml/jax/blob/main/README.md Research reference
- https://github.com/jax-ml/jax/blob/main/LICENSE Research reference
Research metadata
- Research date
- 2026-10-02
- Catalogue snapshot
- 2026-10-06
Community & social 1 channels
Communities
- GitHub Discussionsgithub.com/jax-ml/jax/discussionsOfficial
Link details for GitHub Discussions
- Last checked