tojax

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

Downloads last day: 23
Downloads last week: 278
Downloads last month: 668