Proprietario
rdyro
1 repository indicizzato · Vedi su GitHub
-
torch2jax
Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
Python · 60 stelle
-
Torch2Jax vs Torchax? Aperta
Difficoltà 5/5 Più di una settimana Idoneità per principianti 20/100
-
jax 0.7.0 Breaks Torch2Jax Aperta
Difficoltà 4/5 3-5 giorni Idoneità per principianti 45/100
-
Difficoltà 3/5 1-2 giorni Idoneità per principianti 35/100
-
Difficoltà 3/5 1-2 giorni Idoneità per principianti 35/100
-
Difficoltà 5/5 Più di una settimana Idoneità per principianti 25/100
-
pmap performance Aperta
Difficoltà 4/5 3-5 giorni Idoneità per principianti 35/100
-
Multi-GPU Question Aperta
Difficoltà 4/5 3-5 giorni Idoneità per principianti 25/100
-
Difficoltà 5/5 Più di una settimana Idoneità per principianti 15/100