torchax

github.comtool

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

from

JAX

added

2026-10-10

likes

0

JAX › Libraries > New Libraries: “torchax is a library for Jax to interoperate with model code written in PyTorch.”