Toolchain · 2018
JAX
Array-computing library from Google: NumPy-style API with automatic differentiation and jax.
What is JAX written in?
The JAX compiler is written in XLA (well-documented). Its runtime is implemented in Python and C++ (documented).
Frequently Asked Questions
What language is JAX written in?
JAX is primarily implemented in Python and C++ and XLA.
When was JAX first released?
JAX was first released in 2018.
Evidence Sources
- https://github.com/jax-ml/jax/tree/main/jax Source repository
- https://github.com/jax-ml/jax/tree/main/jaxlib Source repository
- https://github.com/jax-ml/jax/blob/main/README.md Source repository
Discover More
Spot something wrong?
Every fact on this page is meant to be checkable. If one is wrong, say so and it becomes a dataset change, not a patch to this page.