All modules for which code is available
- numpyro.contrib.control_flow.cond
- numpyro.contrib.control_flow.scan
- numpyro.contrib.ecs_proxies
- numpyro.contrib.einstein.stein_kernels
- numpyro.contrib.einstein.steinvi
- numpyro.contrib.funsor.discrete
- numpyro.contrib.funsor.enum_messenger
- numpyro.contrib.funsor.infer_util
- numpyro.contrib.hsgp.approximation
- numpyro.contrib.hsgp.laplacian
- numpyro.contrib.hsgp.spectral_densities
- numpyro.contrib.module
- numpyro.contrib.nested_sampling
- numpyro.contrib.stochastic_support.dcc
- numpyro.contrib.stochastic_support.sdvi
- numpyro.contrib.tfp.distributions
- numpyro.contrib.tfp.mcmc
- numpyro.diagnostics
- numpyro.distributions.conjugate
- numpyro.distributions.constraints
- numpyro.distributions.continuous
- numpyro.distributions.copula
- numpyro.distributions.directional
- numpyro.distributions.discrete
- numpyro.distributions.distribution
- numpyro.distributions.flows
- numpyro.distributions.mixtures
- numpyro.distributions.transforms
- numpyro.distributions.truncated
- numpyro.handlers
- numpyro.infer.autoguide
- numpyro.infer.barker
- numpyro.infer.elbo
- numpyro.infer.ensemble
- numpyro.infer.hmc
- numpyro.infer.hmc_gibbs
- numpyro.infer.hmc_util
- numpyro.infer.initialization
- numpyro.infer.inspect
- numpyro.infer.mcmc
- numpyro.infer.mixed_hmc
- numpyro.infer.reparam
- numpyro.infer.sa
- numpyro.infer.svi
- numpyro.infer.util
- numpyro.ops.indexing
- numpyro.optim
- numpyro.primitives
- numpyro.util