Index

A

acceptance_rate (HMCInfo attribute) (MALAInfo attribute) (MarginalInfo attribute) (NUTSInfo attribute) (RWInfo attribute) adaptation_state (AdaptationInfo attribute) AdaptationAlgorithm (class in blackjax.base) AdaptationInfo (class in blackjax.adaptation.base) AdaptationResults (class in blackjax.adaptation.base) adaptive_tempered_smc (class in blackjax.smc.adaptive_tempered) additive_step_random_walk (class in blackjax.mcmc.random_walk)

alpha (LBFGSHistory attribute) (MEADSAdaptationState attribute) (PathfinderState attribute) ancestors (SMCInfo attribute) append_to_trajectory() (in module blackjax.mcmc.trajectory) approximate (pathfinder attribute) approximate() (in module blackjax.vi.pathfinder) ArrayLikeTree (in module blackjax.types) ArrayTree (in module blackjax.types) asymmetric_proposal_generator() (in module blackjax.mcmc.proposal) avg_error (DualAveragingAdaptationState attribute) (DualAveragingState attribute)

B

base() (in module blackjax.adaptation.meads_adaptation) (in module blackjax.adaptation.pathfinder_adaptation) (in module blackjax.adaptation.window_adaptation) beta (PathfinderState attribute) bfgs_sample() (in module blackjax.optimizers.lbfgs) blackjax module blackjax._version module blackjax.adaptation module blackjax.adaptation.base module blackjax.adaptation.mass_matrix module blackjax.adaptation.meads_adaptation module blackjax.adaptation.pathfinder_adaptation module blackjax.adaptation.step_size module blackjax.adaptation.window_adaptation module blackjax.base module blackjax.diagnostics module blackjax.mcmc module blackjax.mcmc.diffusions module blackjax.mcmc.elliptical_slice module blackjax.mcmc.ghmc module blackjax.mcmc.hmc module blackjax.mcmc.integrators module blackjax.mcmc.mala module blackjax.mcmc.marginal_latent_gaussian module blackjax.mcmc.metrics module blackjax.mcmc.nuts module blackjax.mcmc.periodic_orbital module blackjax.mcmc.proposal module blackjax.mcmc.random_walk module blackjax.mcmc.termination module blackjax.mcmc.trajectory module blackjax.optimizers module blackjax.optimizers.dual_averaging module blackjax.optimizers.lbfgs module blackjax.progress_bar module blackjax.sgmcmc module blackjax.sgmcmc.csgld module

blackjax.sgmcmc.diffusions module blackjax.sgmcmc.gradients module blackjax.sgmcmc.sghmc module blackjax.sgmcmc.sgld module blackjax.sgmcmc.sgnht module blackjax.smc module blackjax.smc.adaptive_tempered module blackjax.smc.base module blackjax.smc.ess module blackjax.smc.resampling module blackjax.smc.solver module blackjax.smc.tempered module blackjax.types module blackjax.util module blackjax.vi module blackjax.vi.meanfield_vi module blackjax.vi.pathfinder module blackjax.vi.svgd module build_additive_step() (in module blackjax.mcmc.random_walk) build_irmh() (in module blackjax.mcmc.random_walk) build_kernel (adaptive_tempered_smc attribute) (additive_step_random_walk attribute) (csgld attribute) (elliptical_slice attribute) (ghmc attribute) (hmc attribute) (irmh attribute) (mala attribute) (nuts attribute) (orbital_hmc attribute) (rmh attribute) (sghmc attribute) (sgld attribute) (sgnht attribute) (svgd attribute) build_kernel() (in module blackjax.mcmc.elliptical_slice) (in module blackjax.mcmc.ghmc) (in module blackjax.mcmc.hmc) (in module blackjax.mcmc.mala) (in module blackjax.mcmc.nuts) (in module blackjax.mcmc.periodic_orbital) (in module blackjax.sgmcmc.csgld) (in module blackjax.sgmcmc.sghmc) (in module blackjax.sgmcmc.sgld) (in module blackjax.sgmcmc.sgnht) (in module blackjax.smc.adaptive_tempered) (in module blackjax.smc.tempered) build_rmh() (in module blackjax.mcmc.random_walk) build_rmh_transition_energy() (in module blackjax.mcmc.random_walk) build_schedule() (in module blackjax.adaptation.window_adaptation)

C

ContourSGLDState (class in blackjax.sgmcmc.csgld) control_variates() (in module blackjax.sgmcmc.gradients)

csgld (class in blackjax.sgmcmc.csgld) current_iteration (MEADSAdaptationState attribute)

D

delta (MEADSAdaptationState attribute) dichotomy() (in module blackjax.smc.solver) directions (PeriodicOrbitalState attribute) dual_averaging() (in module blackjax.optimizers.dual_averaging) dual_averaging_adaptation() (in module blackjax.adaptation.step_size) DualAveragingAdaptationState (class in blackjax.adaptation.step_size)

DualAveragingState (class in blackjax.optimizers.dual_averaging) dynamic_multiplicative_expansion() (in module blackjax.mcmc.trajectory) dynamic_progressive_integration() (in module blackjax.mcmc.trajectory) dynamic_recursive_integration() (in module blackjax.mcmc.trajectory) DynamicExpansionState (class in blackjax.mcmc.trajectory) DynamicIntegrationState (class in blackjax.mcmc.trajectory)

E

effective_sample_size() (in module blackjax.diagnostics) elbo (MFVIInfo attribute) (PathfinderState attribute) EllipSliceInfo (class in blackjax.mcmc.elliptical_slice) EllipSliceState (class in blackjax.mcmc.elliptical_slice) elliptical_slice (class in blackjax.mcmc.elliptical_slice)

energy (HMCInfo attribute) (NUTSInfo attribute) (Proposal attribute) energy_idx (ContourSGLDState attribute) energy_pdf (ContourSGLDState attribute) ess() (in module blackjax.smc.ess) ess_solver() (in module blackjax.smc.ess)

F

f (LBFGSHistory attribute)

find_reasonable_step_size() (in module blackjax.adaptation.step_size)

G

g (LBFGSHistory attribute) gamma (PathfinderState attribute) gaussian_euclidean() (in module blackjax.mcmc.metrics) generate_gaussian_noise() (in module blackjax.util)

generate_meanfield_logdensity() (in module blackjax.vi.meanfield_vi) ghmc (class in blackjax.mcmc.ghmc) GHMCState (class in blackjax.mcmc.ghmc) grad_estimator() (in module blackjax.sgmcmc.gradients) grad_position (PathfinderState attribute)

H

hmc (class in blackjax.mcmc.hmc) hmc_energy() (in module blackjax.mcmc.trajectory)

HMCInfo (class in blackjax.mcmc.hmc) HMCState (class in blackjax.mcmc.hmc)

I

idx_max (IterativeUTurnState attribute) idx_min (IterativeUTurnState attribute) imm_state (WindowAdaptationState attribute) index_pytree() (in module blackjax.util) info (AdaptationInfo attribute) Info (in module blackjax.base) init (adaptive_tempered_smc attribute) (additive_step_random_walk attribute) (csgld attribute) (elliptical_slice attribute) (ghmc attribute) (hmc attribute) (in module blackjax.mcmc.nuts) (irmh attribute) (mala attribute) (meanfield_vi attribute) (nuts attribute) (orbital_hmc attribute) (rmh attribute) (SamplingAlgorithm attribute) (sghmc attribute) (sgld attribute) (sgnht attribute) (svgd attribute) (VIAlgorithm attribute)

init() (in module blackjax.mcmc.elliptical_slice) (in module blackjax.mcmc.ghmc) (in module blackjax.mcmc.hmc) (in module blackjax.mcmc.mala) (in module blackjax.mcmc.periodic_orbital) (in module blackjax.sgmcmc.csgld) (in module blackjax.sgmcmc.sghmc) (in module blackjax.sgmcmc.sgld) (in module blackjax.sgmcmc.sgnht) (in module blackjax.smc.base) (in module blackjax.smc.tempered) init_and_kernel() (in module blackjax.mcmc.marginal_latent_gaussian) InitFn (class in blackjax.base) inverse_mass_matrix (MassMatrixAdaptationState attribute) (PathfinderAdaptationState attribute) (WindowAdaptationState attribute) irmh (class in blackjax.mcmc.random_walk) is_accepted (HMCInfo attribute) (MALAInfo attribute) (MarginalInfo attribute) (RWInfo attribute) is_divergent (HMCInfo attribute) (NUTSInfo attribute) is_turning (NUTSInfo attribute) iterative_uturn_numpyro() (in module blackjax.mcmc.termination) IterativeUTurnState (class in blackjax.mcmc.termination)

L

lbfgs_inverse_hessian_factors() (in module blackjax.optimizers.lbfgs) lbfgs_inverse_hessian_formula_1() (in module blackjax.optimizers.lbfgs) lbfgs_inverse_hessian_formula_2() (in module blackjax.optimizers.lbfgs) LBFGSHistory (class in blackjax.optimizers.lbfgs) leftmost_state (Trajectory attribute) linear_map() (in module blackjax.util) lmbda (TemperedSMCState attribute) log_ess() (in module blackjax.smc.ess) log_likelihood_increment (SMCInfo attribute) log_step_size (DualAveragingAdaptationState attribute) log_step_size_avg (DualAveragingAdaptationState attribute) log_x (DualAveragingState attribute) log_x_avg (DualAveragingState attribute)

logdensities (PeriodicOrbitalState attribute) logdensities_grad (PeriodicOrbitalState attribute) logdensity (EllipSliceState attribute) (GHMCState attribute) (HMCState attribute) (MALAState attribute) (MarginalState attribute) (RWState attribute) logdensity_estimator() (in module blackjax.sgmcmc.gradients) logdensity_grad (GHMCState attribute) (HMCState attribute) (MALAState attribute) (MarginalState attribute)

M

m2 (WelfordAlgorithmState attribute) mala (class in blackjax.mcmc.mala) MALAInfo (class in blackjax.mcmc.mala) MALAState (class in blackjax.mcmc.mala) MarginalInfo (class in blackjax.mcmc.marginal_latent_gaussian) MarginalState (class in blackjax.mcmc.marginal_latent_gaussian) mass_matrix_adaptation() (in module blackjax.adaptation.mass_matrix) MassMatrixAdaptationState (class in blackjax.adaptation.mass_matrix) maximum_eigenvalue() (in module blackjax.adaptation.meads_adaptation) mclachlan() (in module blackjax.mcmc.integrators) meads_adaptation() (in module blackjax.adaptation.meads_adaptation) MEADSAdaptationState (class in blackjax.adaptation.meads_adaptation) mean (WelfordAlgorithmState attribute) meanfield_vi (class in blackjax.vi.meanfield_vi) merge_trajectories() (in module blackjax.mcmc.trajectory) MFVIInfo (class in blackjax.vi.meanfield_vi) MFVIState (class in blackjax.vi.meanfield_vi) mgrad_gaussian (class in blackjax.mcmc.marginal_latent_gaussian) minimize_lbfgs() (in module blackjax.optimizers.lbfgs) module blackjax blackjax._version blackjax.adaptation blackjax.adaptation.base blackjax.adaptation.mass_matrix blackjax.adaptation.meads_adaptation blackjax.adaptation.pathfinder_adaptation blackjax.adaptation.step_size blackjax.adaptation.window_adaptation blackjax.base blackjax.diagnostics blackjax.mcmc blackjax.mcmc.diffusions blackjax.mcmc.elliptical_slice blackjax.mcmc.ghmc blackjax.mcmc.hmc blackjax.mcmc.integrators blackjax.mcmc.mala blackjax.mcmc.marginal_latent_gaussian blackjax.mcmc.metrics blackjax.mcmc.nuts blackjax.mcmc.periodic_orbital blackjax.mcmc.proposal blackjax.mcmc.random_walk blackjax.mcmc.termination blackjax.mcmc.trajectory blackjax.optimizers blackjax.optimizers.dual_averaging blackjax.optimizers.lbfgs blackjax.progress_bar blackjax.sgmcmc blackjax.sgmcmc.csgld blackjax.sgmcmc.diffusions blackjax.sgmcmc.gradients blackjax.sgmcmc.sghmc blackjax.sgmcmc.sgld blackjax.sgmcmc.sgnht blackjax.smc blackjax.smc.adaptive_tempered blackjax.smc.base blackjax.smc.ess blackjax.smc.resampling blackjax.smc.solver blackjax.smc.tempered blackjax.types blackjax.util blackjax.vi blackjax.vi.meanfield_vi blackjax.vi.pathfinder blackjax.vi.svgd

momentum (EllipSliceInfo attribute) (GHMCState attribute) (HMCInfo attribute) (IterativeUTurnState attribute) (NUTSInfo attribute) (SGNHTState attribute) momentum_sum (IterativeUTurnState attribute) (Trajectory attribute) mu (DualAveragingAdaptationState attribute) (DualAveragingState attribute) (MFVIState attribute) multinomial() (in module blackjax.smc.resampling)

N

nonreversible_slice_sampling() (in module blackjax.mcmc.proposal) normal() (in module blackjax.mcmc.random_walk) normal_random_walk() (additive_step_random_walk class method) num_integration_steps (HMCInfo attribute) (NUTSInfo attribute)

num_states (Trajectory attribute) num_trajectory_expansions (NUTSInfo attribute) nuts (class in blackjax.mcmc.nuts) NUTSInfo (class in blackjax.mcmc.nuts)

O

opt_state (MFVIState attribute) orbital_hmc (class in blackjax.mcmc.periodic_orbital)

overdamped_langevin() (in module blackjax.mcmc.diffusions) (in module blackjax.sgmcmc.diffusions)

P

parameters (AdaptationResults attribute) particles (SMCState attribute) (TemperedSMCState attribute) pathfinder (class in blackjax.vi.pathfinder) pathfinder_adaptation() (in module blackjax.adaptation.pathfinder_adaptation) PathfinderAdaptationState (class in blackjax.adaptation.pathfinder_adaptation) PathfinderState (class in blackjax.vi.pathfinder) PeriodicOrbitalState (class in blackjax.mcmc.periodic_orbital) position (ContourSGLDState attribute) (EllipSliceState attribute) (GHMCState attribute) (HMCState attribute) Position (in module blackjax.base) position (MALAState attribute) (MarginalState attribute) (PathfinderState attribute) (RWState attribute) (SGNHTState attribute)

position_sigma (MEADSAdaptationState attribute) positions (PeriodicOrbitalState attribute) potential_scale_reduction() (in module blackjax.diagnostics) PRNGKey (in module blackjax.types) progress_bar_scan() (in module blackjax.progress_bar) progressive_biased_sampling() (in module blackjax.mcmc.proposal) progressive_uniform_sampling() (in module blackjax.mcmc.proposal) Proposal (class in blackjax.mcmc.proposal) proposal (DynamicExpansionState attribute) (DynamicIntegrationState attribute) (HMCInfo attribute) (MarginalInfo attribute) (RWInfo attribute) proposal_from_energy_diff() (in module blackjax.mcmc.proposal) proposal_generator() (in module blackjax.mcmc.proposal) pytree_size() (in module blackjax.util)

R

rbf_kernel() (in module blackjax.vi.svgd) reorder_trajectories() (in module blackjax.mcmc.trajectory) residual() (in module blackjax.smc.resampling) rho (MFVIState attribute) rightmost_state (Trajectory attribute)

rmh (class in blackjax.mcmc.random_walk) rmh_proposal() (in module blackjax.mcmc.random_walk) run (AdaptationAlgorithm attribute) RunFn (class in blackjax.base) RWInfo (class in blackjax.mcmc.random_walk) RWState (class in blackjax.mcmc.random_walk)

S

sample (meanfield_vi attribute) (pathfinder attribute) (VIAlgorithm attribute) sample() (in module blackjax.vi.meanfield_vi) (in module blackjax.vi.pathfinder) sample_size (WelfordAlgorithmState attribute) SamplingAlgorithm (class in blackjax.base) sghmc (class in blackjax.sgmcmc.sghmc) sghmc() (in module blackjax.sgmcmc.diffusions) sgld (class in blackjax.sgmcmc.sgld) sgnht (class in blackjax.sgmcmc.sgnht) sgnht() (in module blackjax.sgmcmc.diffusions) SGNHTState (class in blackjax.sgmcmc.sgnht) slice (GHMCState attribute) SMCInfo (class in blackjax.smc.base) SMCState (class in blackjax.smc.base) ss_state (PathfinderAdaptationState attribute) (WindowAdaptationState attribute) state (AdaptationInfo attribute) (AdaptationResults attribute)

State (in module blackjax.base) state (Proposal attribute) static_binomial_sampling() (in module blackjax.mcmc.proposal) static_integration() (in module blackjax.mcmc.trajectory) step (DualAveragingAdaptationState attribute) (DualAveragingState attribute) (DynamicExpansionState attribute) (DynamicIntegrationState attribute) (meanfield_vi attribute) (SamplingAlgorithm attribute) (VIAlgorithm attribute) step() (in module blackjax.smc.base) (in module blackjax.vi.meanfield_vi) step_size (MEADSAdaptationState attribute) (PathfinderAdaptationState attribute) (WindowAdaptationState attribute) stratified() (in module blackjax.smc.resampling) subiter (EllipSliceInfo attribute) sum_log_p_accept (Proposal attribute) svgd (class in blackjax.vi.svgd) systematic() (in module blackjax.smc.resampling)

T

TemperedSMCState (class in blackjax.smc.tempered) termination_state (DynamicExpansionState attribute) (DynamicIntegrationState attribute) theta (EllipSliceInfo attribute) Trajectory (class in blackjax.mcmc.trajectory)

trajectory (DynamicExpansionState attribute) (DynamicIntegrationState attribute) trajectory_leftmost_state (NUTSInfo attribute) trajectory_rightmost_state (NUTSInfo attribute) TrajectoryState (in module blackjax.mcmc.proposal)

U

U_grad_x (MarginalState attribute) U_x (MarginalState attribute) update_info (SMCInfo attribute)

update_mask (LBFGSHistory attribute) update_median_heuristic() (in module blackjax.vi.svgd) UpdateFn (class in blackjax.base)

V

velocity_verlet() (in module blackjax.mcmc.integrators)

VIAlgorithm (class in blackjax.base)

W

wc_state (MassMatrixAdaptationState attribute) weight (Proposal attribute) weights (PeriodicOrbitalState attribute) (SMCState attribute) (TemperedSMCState attribute)

welford_algorithm() (in module blackjax.adaptation.mass_matrix) WelfordAlgorithmState (class in blackjax.adaptation.mass_matrix) window_adaptation() (in module blackjax.adaptation.window_adaptation) WindowAdaptationState (class in blackjax.adaptation.window_adaptation)

X

x (LBFGSHistory attribute)

xi (SGNHTState attribute)

Y

yoshida() (in module blackjax.mcmc.integrators)