Illustration dessinée par RepoMasters à partir de l’identifiant du dépôt et de la couleur de son langage. Crédits
PrototypeCommune

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.

Propriétaire
google (organisation)
Étoiles
Étoiles : 249
Forks
Forks : 41
Langage
Python
Licence
Apache-2.0
Création
2025
Sujets
aiinteroperabilityjaxmachine-learningpytorchtpu
Rang
Prototype (100 à 249 ★) Ce que veut dire ce rang
Voir sur GitHub

Autres dépôts de google

Les 1 050 dépôts de google

D’autres dépôts Python

Tous les dépôts Python