numpyro (conda): Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Keywords: bayesian-inference|hmc|inference-algorithms|jax|mcmc|numpy|probabilistic-programming|pyro