numpyro (pypi): Pyro PPL on NumPy
Keywords: probabilistic|machine|learning|bayesian|statistics|bayesian-inference|hmc|inference-algorithms|jax|mcmc|numpy|probabilistic-programming|pyro