Package Usage: conda: jax
JAX is Autograd and XLA, brought together for high-performance numerical computing and machine learning research. It provides composable transformations of Python+NumPy programs: differentiate, vectorize, parallelize, Just-In-Time compile to GPU/TPU, and more.
3 versions
Latest release: over 1 year ago
6 dependent packages
View more package details: https://packages.ecosystem.code.gouv.fr/registries/anaconda.org/packages/jax