NumPyro
v0.2.2
Modeling:
Pyro Primitives
Effect Handlers
Distributions:
Base Distribution
Continuous Distributions
Discrete Distributions
Constraints
Transforms
Flows
Inference:
Markov Chain Monte Carlo (MCMC)
Stochastic Variational Inference (SVI)
Automatic Guide Generation
Optimizers
Diagnostics
Runtime Utilities
Inference Utilities
NumPyro
Docs
»
NumPyro documentation
Edit on GitHub
NumPyro documentation
¶
Modeling:
Pyro Primitives
param
sample
plate
factor
module
Effect Handlers
block
condition
replay
scale
seed
substitute
trace
Distributions:
Base Distribution
Distribution
Independent
TransformedDistribution
Unit
Continuous Distributions
Beta
Cauchy
Chi2
Dirichlet
Exponential
Gamma
GaussianRandomWalk
HalfCauchy
HalfNormal
InverseGamma
LKJ
LKJCholesky
LogNormal
MultivariateNormal
LowRankMultivariateNormal
Normal
Pareto
StudentT
TruncatedCauchy
TruncatedNormal
Uniform
Discrete Distributions
Bernoulli
BernoulliLogits
BernoulliProbs
BetaBinomial
Binomial
BinomialLogits
BinomialProbs
Categorical
CategoricalLogits
CategoricalProbs
Delta
GammaPoisson
Multinomial
MultinomialLogits
MultinomialProbs
OrderedLogistic
Poisson
PRNGIdentity
ZeroInflatedPoisson
Constraints
boolean
corr_cholesky
corr_matrix
dependent
greater_than
integer_interval
integer_greater_than
interval
lower_cholesky
multinomial
nonnegative_integer
ordered_vector
positive
positive_definite
positive_integer
real
real_vector
simplex
unit_interval
Transforms
biject_to
Transform
AbsTransform
AffineTransform
ComposeTransform
CorrCholeskyTransform
ExpTransform
IdentityTransform
InvCholeskyTransform
LowerCholeskyTransform
MultivariateAffineTransform
OrderedTransform
PermuteTransform
PowerTransform
SigmoidTransform
StickBreakingTransform
Flows
InverseAutoregressiveTransform
Inference:
Markov Chain Monte Carlo (MCMC)
Hamiltonian Monte Carlo
MCMC Utilities
Stochastic Variational Inference (SVI)
ELBO
RenyiELBO
Automatic Guide Generation
AutoContinuous
AutoDiagonalNormal
AutoMultivariateNormal
AutoIAFNormal
AutoLaplaceApproximation
AutoContinuousELBO
Optimizers
Adam
Adagrad
ClippedAdam
Momentum
RMSProp
RMSPropMomentum
SGD
SM3
Diagnostics
Autocorrelation
Autocovariance
Effective Sample Size
Gelman Rubin
Split Gelman Rubin
HPDI
Summary
Runtime Utilities
enable_validation
validation_enabled
enable_x64
set_platform
set_host_device_count
Inference Utilities
Predictive
log_density
transform_fn
constrain_fn
potential_energy
log_likelihood
find_valid_initial_params
Initialization Strategies
Indices and tables
¶
Index
Search Page
Read the Docs
v: v0.2.2
Versions
latest
stable
v0.2.2
v0.2.1
v0.2.0
v0.1.0
Downloads
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.