
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 hardware accelerated vectorization.