wrap-torch2jax 0.9.0

T1Covered by 1 source · first reported by PyPI / crates.io / RubyGems / Go index / NuGet at 26 Sept, 06:31 UTC

Wrap your PyTorch for JAX! This package allows no-copy PyTorch calling from JAX under both eager execution and JIT.