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