JAX
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU.
Just now
Has API
PricingFree
Free
Automatic Differentiation
XLA Compilation
Hardware-accelerated Vectorization
AI-Ranked Results
Automatic Differentiation Tools
Discover the strongest tools and workflows for automatic differentiation.
1 tools foundFree tools displayed firstSorted by rating
Filter Results
Matching Tools (1)
AI Framework