Package Usage: pypi: pytorch2jax
Convert PyTorch models to Jax functions and Flax models
4 versions
Latest release: over 2 years ago
692 downloads last month
View more package details: https://packages.ecosystem.code.gouv.fr/registries/pypi.org/packages/pytorch2jax