JAX
Python library for composable transformations of NumPy-style code: autodiff, JIT compilation and vectorization.
Overview
Python library for composable transformations of NumPy-style code: autodiff, JIT compilation and vectorization.
- Category
- Machine learning frameworks · AI tools
- Platforms
- Linux, macOS, API only
- Pricing
- Free
- License
- Apache-2.0
- Source code
- jax-ml/jax
- Repository stars
- 36,314
- API
- No public API description
Release history
Updated 2 days ago10 releases in the last 12 months.
-
jax-v0.11.2
-
jax-v0.11.1
-
jax-v0.11.0
-
jax-v0.10.2
-
jax-v0.10.1
-
jax-v0.10.0
What users say
No reviews yet
Ratings appear once verified reviews arrive. Until then the rank uses public signals only: release freshness and repository adoption.
Alternatives to JAX
See all alternativesPython library with model definitions and pretrained weights for text, vision and audio transformers.
Python deep learning framework with tensors, automatic differentiation and GPU acceleration.
Python library for classical machine learning: classification, regression, clustering and preprocessing.
Machine learning platform for building, training and deploying models on servers, browsers and devices.