numpyro
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
2k stars
52 watching
235 forks
Language: Python
last commit: 4 days ago
Linked from 1 awesome list
bayesian-inferencehmcinference-algorithmsjaxmcmcnumpyprobabilistic-programmingpyro
n2cholas/awesome-jax