numpyro
Probabilistic compiler
A probabilistic programming library that provides a NumPy backend for automatic differentiation and JIT compilation to GPU/TPU/CPU.
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
2k stars
53 watching
244 forks
Language: Python
last commit: 4 days ago
Linked from 1 awesome list
bayesian-inferencehmcinference-algorithmsjaxmcmcnumpyprobabilistic-programmingpyro
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 probprog/pyprob | A probabilistic programming system for simulators and high-performance computing based on PyTorch | 27 |
 probtorch/probtorch | A deep learning framework for probabilistic models that extends PyTorch with support for reparameterized distributions and Monte Carlo estimators. | 887 |
 rlouf/mcx | Tools and methods for Bayesian deep learning using probabilistic programming. | 325 |
 stepelu/ptstat | A PyTorch-based implementation of probabilistic programming and statistical inference tools | 109 |
 camdavidsonpilon/probabilistic-programming-and-bayesian-methods-for-hackers | An introduction to Bayesian methods and probabilistic programming for software developers | 26,798 |
 google/edward2 | A tool for writing probabilistic models and manipulating their computation | 679 |
 pgm-lab/inferpy | A high-level API for probabilistic modeling with a focus on ease of use and scalability | 147 |
 thepaw/probab | A tool for generating and manipulating probabilistic models in code. | 20 |
 alexis-jacq/pytorch-dppo | A PyTorch implementation of Distributed Proximal Policy Optimization algorithm | 180 |
 pyos/dg | A programming language compiler for CPython bytecode | 576 |
 aalhour/pycoolc | A compiler for the COOL programming language targeting the MIPS 32-bit architecture and implemented in Python 3 | 40 |
 pymc-devs/pymc | A Python package for Bayesian statistical modeling and probabilistic programming | 8,722 |
 ctallec/pyvarinf | A Python package facilitating Bayesian Deep Learning methods with Variational Inference for PyTorch | 359 |
 probcomp/bayeslite | A database system that allows querying probabilistic implications of data using Bayesian inference. | 923 |
 uber/orbit | A Python package for Bayesian time series forecasting and inference with probabilistic programming languages under the hood. | 1,883 |