JAXopt

github.comtool

Hardware accelerated (GPU/TPU), batchable and differentiable optimizers in JAX.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲

   JAX Toolbox github.com

   Nightly CI and optimized examples for JAX on NVIDIA GPUs using libraries such as T5x, Paxml, and Transformer Engine.

   tool topic ai link ↗
   
2. ▲

   SymJAX github.com

   Symbolic CPU/GPU/TPU programming.

   tool topic programming link ↗
   
3. ▲

   OTT-JAX github.com

   Optimal transport tools in JAX.

   tool link ↗
   
4. ▲

   JAX github.com

   Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.

   tool topic programming link ↗
   
5. ▲

   GPJax github.com

   Gaussian processes in JAX.

   tool link ↗
   
6. ▲

   mpi4jax github.com

   Combine MPI operations with your Jax code on CPUs and GPUs.

   tool link ↗
   

JAX › Libraries > New Libraries: “Hardware accelerated (GPU/TPU), batchable and differentiable optimizers in JAX.”