Objax

github.comtool

Has an object oriented design similar to PyTorch.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   torchax github.com

   tool category #jax link ↗

   torchax is a library for Jax to interoperate with model code written in PyTorch.

   
2. ▲
   FedJAX github.com

   tool category #jax link ↗

   Federated learning in JAX, built on Optax and Haiku.

   
3. ▲
   Pax github.com

   tool category #jax link ↗

   A Jax-based machine learning framework for training large scale models.

4. ▲
   JAXopt github.com

   tool category #jax link ↗

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

   
5. ▲
   jax github.com

   tool category #python link ↗
6. ▲
   OTT-JAX github.com

   tool category #jax link ↗

   Optimal transport tools in JAX.

   

JAX › Libraries: “Has an object oriented design similar to PyTorch.”