JAX

Python library for composable transformations of NumPy-style code: autodiff, JIT compilation and vectorization.

by Google No reviews yet Open source Free jax-v0.11.2 Updated 2 days ago 69 88.0 =

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 ago

10 releases in the last 12 months.

  1. jax-v0.11.2
  2. jax-v0.11.1
  3. jax-v0.11.0
  4. jax-v0.10.2
  5. jax-v0.10.1
  6. 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 alternatives