pytorch2jax

PyPI page
Home page
Author: Subhojeet Pramanik
License:
Summary: Convert PyTorch models to Jax functions and Flax models
Latest version: 0.1.0
Required dependencies: jax | jaxlib | torch

Downloads last day: 2
Downloads last week: 186
Downloads last month: 2,325