NumPyro
0.3.0
Modeling:
Pyro Primitives
Effect Handlers
Distributions:
Base Distribution
Continuous Distributions
Discrete Distributions
Directional Distributions
Constraints
Transforms
Flows
Inference:
Markov Chain Monte Carlo (MCMC)
Stochastic Variational Inference (SVI)
Automatic Guide Generation
Reparameterizers
Funsor-based NumPyro
Optimizers
Diagnostics
Runtime Utilities
Inference Utilities
NumPyro
Docs
»
NumPyro documentation
Edit on GitHub
NumPyro documentation
¶
Modeling:
Pyro Primitives
param
sample
plate
plate_stack
deterministic
factor
module
scan
Effect Handlers
block
condition
lift
mask
reparam
replay
scale
scope
seed
substitute
trace
Distributions:
Base Distribution
Distribution
ExpandedDistribution
ImproperUniform
Independent
MaskedDistribution
TransformedDistribution
Unit
Continuous Distributions
Beta
Cauchy
Chi2
Dirichlet
Exponential
Gamma
Gumbel
GaussianRandomWalk
HalfCauchy
HalfNormal
InverseGamma
Laplace
LKJ
LKJCholesky
LogNormal
Logistic
MultivariateNormal
LowRankMultivariateNormal
Normal
Pareto
StudentT
TruncatedCauchy
TruncatedNormal
TruncatedPolyaGamma
Uniform
Discrete Distributions
Bernoulli
BernoulliLogits
BernoulliProbs
BetaBinomial
Binomial
BinomialLogits
BinomialProbs
Categorical
CategoricalLogits
CategoricalProbs
Delta
GammaPoisson
Multinomial
MultinomialLogits
MultinomialProbs
OrderedLogistic
Poisson
PRNGIdentity
ZeroInflatedPoisson
Directional Distributions
VonMises
Constraints
Constraint
boolean
corr_cholesky
corr_matrix
dependent
greater_than
integer_interval
integer_greater_than
interval
less_than
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
LowerCholeskyAffine
LowerCholeskyTransform
OrderedTransform
PermuteTransform
PowerTransform
SigmoidTransform
StickBreakingTransform
Flows
InverseAutoregressiveTransform
BlockNeuralAutoregressiveTransform
Inference:
Markov Chain Monte Carlo (MCMC)
Hamiltonian Monte Carlo
MCMC Utilities
Stochastic Variational Inference (SVI)
ELBO
RenyiELBO
Automatic Guide Generation
AutoContinuous
AutoBNAFNormal
AutoDiagonalNormal
AutoMultivariateNormal
AutoIAFNormal
AutoLaplaceApproximation
AutoLowRankMultivariateNormal
Reparameterizers
Loc-Scale Decentering
Neural Transport
Transformed Distributions
Funsor-based NumPyro
Effect handlers
Inference Utilities
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
Tensor Indexing
Indices and tables
¶
Index
Search Page
Read the Docs
v: 0.3.0
Versions
latest
stable
0.3.0
v0.2.4
v0.2.3
v0.2.2
v0.2.1
v0.2.0
v0.1.0
fix-docs
Downloads
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.