PrototypeCommon
google/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.
- Owner
- google (organization)
- Stars
- Stars: 249
- Forks
- Forks: 41
- Language
- Python
- License
- Apache-2.0
- Created
- 2025
- Topics
- aiinteroperabilityjaxmachine-learningpytorchtpu
- Tier
- Prototype (100 to 249 ★) What this tier means
More repos by google
All 1,050 repos by google- google/material-design-iconsStars: 54,214LTS
- google/guavaStars: 51,915JavaLTS
- google/zxStars: 45,832JavaScriptLTS
- google/styleguideStars: 39,788Emacs LispLTS
- google/googletestStars: 39,690C++LTS
- google/leveldbStars: 39,516C++LTS
Nearby in Python
All Python repos- bbaudio-2025/Comfyui-MMH3-UltimateUpscaleStars: 249PythonPrototype
- ceyyy427/finathinkStars: 249PythonPrototype
- arpanjot-coding/DatlyStars: 249PythonPrototype
- braintree/braintree_pythonStars: 249PythonPrototype
- JulianEberius/SublimeRopeStars: 249PythonPrototype
- firecat53/urlscanStars: 249PythonPrototype