samuela / torch2jax

Run PyTorch in JAX. 🤝
199Updated last year

Related projects

Alternatives and complementary repositories for torch2jax