samuela / torch2jax

Run PyTorch in JAX. 🤝
200Updated last year

Related projects

Alternatives and complementary repositories for torch2jax