issues
search
rdyro
/
torch2jax
Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
https://rdyro.github.io/torch2jax/
MIT License
37
stars
1
forks
source link
Ignore
#5
Closed
adam-hartshorne
closed
1 year ago
adam-hartshorne
commented
1 year ago
Ignore
Ignore