MarketplaceMachine Learning Model TrainingTorchax

Torchax

byGoogle

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…

Machine Learning Model TrainingML FrameworksOpen Source
Visita il sito web

About

Pagina della documentazione: https://google.github.io/torchax/ Canale di discussione Discord: https://discord.gg/JqeJqGPyzC

Open Source Health

Not enough history
Stars
243
Forks
38
License
Apache-2.0
Last commit
1 months ago
Python

Alternative a Torchax

Resources & Links

Website

github.com/google/torchax

Related Categories

Machine Learning Model TrainingML Frameworks

Usi Torchax?

Tieni traccia del suo costo insieme al resto del tuo stack e ricevi un promemoria prima del rinnovo.

Aggiungi al mio stack

Vendor

Google

Organize the world's information and make it universally accessible.

View vendor profile

More by Google

Corpuscrawler

Skills

Or Tools

Jetpack Camera App