rdyro / torch2jaxLinks

Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
53Updated last month

Alternatives and similar repositories for torch2jax

Users that are interested in torch2jax are comparing it to the libraries listed below

Sorting: