JAX XLA PJRT Plugin for NVIDIA GPUs
No known vulnerabilities in the latest version
Based on latest version 0.9.1. If you're running an older version, check OSV for your specific version.
[](/packages/jax-cuda12-pjrt)
<a href="/packages/jax-cuda12-pjrt"><img src="/api/badges/jax-cuda12-pjrt?period=month" alt="PyPI Stats"></a>