
JAX
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU.
5d ago
Best for Numerical ComputingHas API
PricingFree
Free
Automatic Differentiation
XLA Compilation
Hardware-accelerated Vectorization
Discover the strongest tools and workflows for distributed model training.

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU.