Parallax

github.comtool

Immutable Torch Modules for JAX.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   Eqxvision github.com

   tool category #jax link ↗

   Equinox version of Torchvision.

2. ▲
   torchax github.com

   tool category #jax link ↗

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

   
3. ▲
   jparallax github.com

   tool category #javascript link ↗
   
4. ▲
   FlaxVision github.com

   tool category #jax link ↗

   Flax version of TorchVision.

   
5. ▲
   Lorax github.com

   tool category #jax link ↗

   Automatically apply LoRA to JAX models (Flax, Haiku, etc.)

   
6. ▲
   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.

   

JAX › Libraries > New Libraries: “Immutable Torch Modules for JAX.”