OTT-JAX

github.comtool

Optimal transport tools in JAX.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   Optimal Transport Tools github.com

   tool category #jax link ↗

   Toolbox that bundles utilities to solve optimal transport problems.

   
2. ▲
   JAXopt github.com

   tool category #jax link ↗

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

   
3. ▲
   GPJax github.com

   tool category #jax link ↗

   Gaussian processes in JAX.

   
4. ▲
   TF2JAX github.com

   tool category #jax link ↗

   Convert functions/graphs to JAX functions.

   
5. ▲
   JAX Toolbox github.com

   tool category #jax link ↗

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

   
6. ▲
   FedJAX github.com

   tool category #jax link ↗

   Federated learning in JAX, built on Optax and Haiku.

   

JAX › Libraries > New Libraries: “Optimal transport tools in JAX.”