Rare find

torch2jax. Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.

github.com/rdyro/torch2jax
by @rdyro

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.