tojax

PyPI page
Home page
Author: None
Summary: Effortlessly transform PyTorch functions to JAX
Latest version: 1.0.0
Required dependencies: flax | jax | torch

Downloads last day: 18
Downloads last week: 77
Downloads last month: 352