jax-ml / jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
- Repository stars
- 36,569All-time #890
- Fork
- 4,016
- Last push
- updated 3 days ago
- Repository created
- since 2018
About
Summarised by AI from the README; the original prevails
Author description: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Basic listing based on official public information; functionality has not been independently evaluated.
- Platform
- Not verified
- Pricing
- Pricing not verifiedCheck current official software, hosting, model and infrastructure pricing.
- Maintenance notes
- Not verified