JAX
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.
- from
- Machine Learning, Data Science, Data Science
- added
- 2026-10-10
- likes
- 0
Machine Learning › Python > General-Purpose Machine Learning: “JAX is Autograd and XLA, brought together for high-performance machine learning research.”
Data Science › Deep Learning > JAX: “Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.”