https://github.com/pyro-ppl/numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
https://github.com/pyro-ppl/numpyro

bayesian-inference hmc inference-algorithms jax mcmc numpy probabilistic-programming pyro

Last synced: about 2 months ago
JSON representation

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

• Host: GitHub
• URL: https://github.com/pyro-ppl/numpyro
• Owner: pyro-ppl
• License: apache-2.0
• Created: 2019-02-13T21:13:59.000Z (over 5 years ago)
• Default Branch: master
• Last Pushed: 2024-05-14T16:39:56.000Z (4 months ago)
• Last Synced: 2024-05-15T15:50:16.266Z (4 months ago)
• Topics: bayesian-inference, hmc, inference-algorithms, jax, mcmc, numpy, probabilistic-programming, pyro
• Language: Python
• Homepage: https://num.pyro.ai
• Size: 57.7 MB
• Stars: 2,057
• Watchers: 49
• Forks: 220
• Open Issues: 73
• Metadata Files:
  • Readme: README.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE.md

Awesome Lists containing this project

• awesome-jax - NumPyro - Probabilistic programming based on the Pyro library. <img src="https://img.shields.io/github/stars/pyro-ppl/numpyro?style=social" align="center"> (Libraries)