pyro-ppl/numpyro
Probabilistic programming library that provides a NumPy backend for Pyro using JAX for automatic differentiation and GPU/TPU compilation.
Not currently ranked — collecting fresh signals.
star history
NumPyro is a lightweight probabilistic programming framework built on JAX for high-performance numerical computation. It implements inference algorithms with a focus on MCMC methods like Hamiltonian Monte Carlo (HMC) and NUTS, along with variational inference support. The library enables users to write probabilistic models using NumPy-compatible syntax while leveraging JAX’s automatic differentiation and just-in-time compilation across hardware accelerators.
Frequently asked
- What is pyro-ppl/numpyro?
- Probabilistic programming library that provides a NumPy backend for Pyro using JAX for automatic differentiation and GPU/TPU compilation.
- Is numpyro open source?
- Yes — pyro-ppl/numpyro is open source, released under the Apache-2.0 license.
- What language is numpyro written in?
- pyro-ppl/numpyro is primarily written in Python.
- How popular is numpyro?
- pyro-ppl/numpyro has 2.7k stars on GitHub.
- Where can I find numpyro?
- pyro-ppl/numpyro is on GitHub at https://github.com/pyro-ppl/numpyro.