torchax is a library for running Jax and PyTorch together
No known vulnerabilities in the latest version
[](/packages/torchax)
<a href="/packages/torchax"><img src="/api/badges/torchax?period=month" alt="PyPI Stats"></a>