Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
No known vulnerabilities in the latest version
Based on latest version 0.20.1. If you're running an older version, check OSV for your specific version.
[](/packages/numpyro)
<a href="/packages/numpyro"><img src="/api/badges/numpyro?period=month" alt="PyPI Stats"></a>