subho406 / pytorch2jax

Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.
19Updated last year

Related projects

Alternatives and complementary repositories for pytorch2jax