pyro-ppl/numpyro

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

[view on github]last commit: Jun 8, 2026

stars

2,707

7d

+4

30d

+24

90d

+80

## star history

## found in

Awesome Open Source AI/Deep Learning Frameworks