All modules for which code is available

• numpyro.contrib.control_flow.cond
• numpyro.contrib.control_flow.scan
• numpyro.contrib.funsor.discrete
• numpyro.contrib.funsor.enum_messenger
• numpyro.contrib.funsor.infer_util
• numpyro.contrib.indexing
• numpyro.contrib.module
• numpyro.contrib.nested_sampling
• numpyro.contrib.optim
• numpyro.contrib.render
• numpyro.contrib.tfp.distributions
• numpyro.contrib.tfp.mcmc
• numpyro.diagnostics
• numpyro.distributions.conjugate
• numpyro.distributions.constraints
• numpyro.distributions.continuous
• numpyro.distributions.directional
• numpyro.distributions.discrete
• numpyro.distributions.distribution
• numpyro.distributions.flows
• numpyro.distributions.transforms
• numpyro.distributions.truncated
• numpyro.distributions.util
• numpyro.handlers
• numpyro.infer.autoguide
• numpyro.infer.barker
• numpyro.infer.elbo
• numpyro.infer.hmc
• numpyro.infer.hmc_gibbs
• numpyro.infer.hmc_util
• numpyro.infer.initialization
• numpyro.infer.mcmc
• numpyro.infer.mixed_hmc
• numpyro.infer.reparam
• numpyro.infer.sa
• numpyro.infer.svi
• numpyro.infer.util
• numpyro.optim
• numpyro.primitives
• numpyro.util