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