jax2torch (pypi): Jax 2 Torch
Keywords: jax|pytorch|deep-learning-framework|torch