Rare find

torchax. torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JAX-Pytorch interoperability, meaning, one can mix JAX & Pytorch syntax together when authoring ML programs, and run it in every hardware JAX can run.

github.com/google/torchax

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.