Numpyro documentation

Inference:

• Markov Chain Monte Carlo (MCMC)
  • Hamiltonian Monte Carlo
  • MCMC Utilities
• Stochastic Variational Inference (SVI)
  • ELBo

Distributions:

• Base Distribution
  • Distribution
  • TransformedDistribution
• Continuous Distributions
  • Beta
  • Cauchy
  • Chi2
  • Dirichlet
  • Exponential
  • Gamma
  • GaussianRandomWalk
  • HalfCauchy
  • HalfNormal
  • LKJCholesky
  • LogNormal
  • Normal
  • Pareto
  • StudentT
  • TruncatedCauchy
  • TruncatedNormal
  • Uniform
• Discrete Distributions
  • Bernoulli
  • BernoulliLogits
  • BernoulliProbs
  • Binomial
  • BinomialLogits
  • BinomialProbs
  • Categorical
  • CategoricalLogits
  • CategoricalProbs
  • Multinomial
  • MultinomialLogits
  • MultinomialProbs
  • Poisson

Inference Primitives:

• Pyro Primitives
• Effect Handlers

Diagnostics:

• Autocorrelation
• Autocovariance
• Effective Sample Size
• Gelman Rubin
• Split Gelman Rubin
• HPDI
• Summary

Indices and tables

• Index
• Search Page