AI Infrastructure SoftwareJAX development team
JAX goes open source
Open-sourced JAX
Why it matters
With composable grad/jit/vmap transformations and XLA compilation, it formed the standard route for large-scale training on TPUs. The repository states that it is a research project, not an official Google product
Source — Primary sources first; secondary reports are labelled “Press”
Officialgithub.comhttps://github.com/jax-ml/jaxAdded in Week 35, 2026 · See that week