pyro-ppl/numpyro

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

语言：Python · GitHub 星标：2,774 · Fork：338

话题：bayesian-inference · hmc · inference-algorithms · jax · mcmc · numpy · probabilistic-programming · pyro

在 GitHub 上查看原仓库