Index A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | V | W | X | Y A a (StepSizeTuningParameter attribute) acceptance_rate (BarkerInfo attribute) (GISTInfo attribute) (GISTStepSizeInfo attribute) (GISTTrajectoryLengthInfo attribute) (HMCInfo attribute) (LaplaceHMCInfo attribute) (MALAInfo attribute) (MarginalInfo attribute) (NUTSInfo attribute) (RWInfo attribute) accumulating_split_pop_buffer() (in module blackjax.adaptation.metric_buffers) AccumulatingSplitPopState (class in blackjax.adaptation.metric_buffers) Adaptation (class in blackjax.adaptation.laps) (class in blackjax.adaptation.laps_burn_in) adaptation_state (AdaptationInfo attribute) AdaptationInfo (class in blackjax.adaptation.base) AdaptationResults (class in blackjax.adaptation.base) AdaptationState (class in blackjax.adaptation.laps) (class in blackjax.adaptation.laps_burn_in) adaptive_persistent_sampling_smc (in module blackjax) adaptive_tempered_smc (in module blackjax) add_all_chains_info() (in module blackjax.eca) add_splitR() (in module blackjax.eca) additive_step_random_walk (in module blackjax) additive_step_random_walk() (in module blackjax.mcmc.random_walk) adjusted_mclmc (in module blackjax) adjusted_mclmc_dynamic (in module blackjax) adjusted_mclmc_find_L_and_step_size() (in module blackjax.adaptation.adjusted_mclmc_adaptation) adjusted_mclmc_make_adaptation_L() (in module blackjax.adaptation.adjusted_mclmc_adaptation) adjusted_mclmc_make_L_step_size_adaptation() (in module blackjax.adaptation.adjusted_mclmc_adaptation) alpha (Adaptation attribute) (MEADSAdaptationState attribute) (PathfinderState attribute) ancestors (PersistentStateInfo attribute) (SMCInfo attribute) append_to_trajectory() (in module blackjax.mcmc.trajectory) approximate() (in module blackjax.vi.pathfinder) as_top_level_api() (in module blackjax.mcmc.adjusted_mclmc) (in module blackjax.mcmc.adjusted_mclmc_dynamic) (in module blackjax.mcmc.barker) (in module blackjax.mcmc.dynamic_hmc) (in module blackjax.mcmc.elliptical_slice) (in module blackjax.mcmc.ghmc) (in module blackjax.mcmc.gist) (in module blackjax.mcmc.gist_step_size) (in module blackjax.mcmc.gist_trajectory_length) (in module blackjax.mcmc.hmc) (in module blackjax.mcmc.laplace_dynamic_hmc) (in module blackjax.mcmc.laplace_hmc) (in module blackjax.mcmc.mala) (in module blackjax.mcmc.marginal_latent_gaussian) (in module blackjax.mcmc.mclmc) (in module blackjax.mcmc.nuts) (in module blackjax.mcmc.periodic_orbital) (in module blackjax.mcmc.rmhmc) (in module blackjax.mcmc.slice) (in module blackjax.ns.nss) (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_persistent_sampling) (in module blackjax.smc.adaptive_tempered) (in module blackjax.smc.inner_kernel_tuning) (in module blackjax.smc.partial_posteriors_path) (in module blackjax.smc.persistent_sampling) (in module blackjax.smc.pretuning) (in module blackjax.smc.tempered) (in module blackjax.vi.fullrank_vi) (in module blackjax.vi.meanfield_vi) (in module blackjax.vi.multipathfinder) (in module blackjax.vi.pathfinder) (in module blackjax.vi.svgd) avg_error (DualAveragingAdaptationState attribute) B b (StepSizeTuningParameter attribute) background_split (LowRankAdaptationState attribute) (LowRankMetricCoreState attribute) barker (in module blackjax) barker_proposal (in module blackjax) BarkerInfo (class in blackjax.mcmc.barker) BarkerState (class in blackjax.mcmc.barker) base() (in module blackjax.adaptation.chees_adaptation) (in module blackjax.adaptation.meads_adaptation) (in module blackjax.adaptation.pathfinder_adaptation) (in module blackjax.adaptation.window_adaptation) beta (PathfinderState attribute) bias() (in module blackjax.adaptation.laps) bias_type (Adaptation attribute) blackjax module blackjax.adaptation module blackjax.adaptation.adjusted_mclmc_adaptation module blackjax.adaptation.base module blackjax.adaptation.chees_adaptation module blackjax.adaptation.laps module blackjax.adaptation.laps_burn_in module blackjax.adaptation.low_rank_adaptation module blackjax.adaptation.mass_matrix module blackjax.adaptation.mclmc_adaptation module blackjax.adaptation.mclmc_lrd_adaptation module blackjax.adaptation.meads_adaptation module blackjax.adaptation.meta module blackjax.adaptation.meta.builders module blackjax.adaptation.meta.verdict module blackjax.adaptation.meta_adaptation module blackjax.adaptation.metric_buffers module blackjax.adaptation.metric_estimators module blackjax.adaptation.metric_recipes module blackjax.adaptation.pathfinder_adaptation module blackjax.adaptation.staged_adaptation module blackjax.adaptation.step_size module blackjax.adaptation.window_adaptation module blackjax.diagnostics module blackjax.eca module blackjax.mcmc module blackjax.mcmc.adjusted_mclmc module blackjax.mcmc.adjusted_mclmc_dynamic module blackjax.mcmc.barker module blackjax.mcmc.diffusions module blackjax.mcmc.dynamic_hmc module blackjax.mcmc.elliptical_slice module blackjax.mcmc.ghmc module blackjax.mcmc.gist module blackjax.mcmc.gist_step_size module blackjax.mcmc.gist_trajectory_length module blackjax.mcmc.hmc module blackjax.mcmc.integrators module blackjax.mcmc.laplace_dynamic_hmc module blackjax.mcmc.laplace_hmc module blackjax.mcmc.laplace_marginal module blackjax.mcmc.mala module blackjax.mcmc.marginal_latent_gaussian module blackjax.mcmc.mclmc 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.rmhmc module blackjax.mcmc.slice module blackjax.mcmc.termination module blackjax.mcmc.trajectory module blackjax.ns module blackjax.ns.adaptive module blackjax.ns.base module blackjax.ns.from_mcmc module blackjax.ns.integrator module blackjax.ns.nss module blackjax.ns.utils module blackjax.progress_reader 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_persistent_sampling module blackjax.smc.adaptive_tempered module blackjax.smc.base module blackjax.smc.ess module blackjax.smc.from_mcmc module blackjax.smc.inner_kernel_tuning module blackjax.smc.partial_posteriors_path module blackjax.smc.persistent_sampling module blackjax.smc.pretuning module blackjax.smc.resampling module blackjax.smc.solver module blackjax.smc.tempered module blackjax.smc.tuning module blackjax.smc.tuning.from_kernel_info module blackjax.smc.tuning.from_particles module blackjax.smc.waste_free module blackjax.vi module blackjax.vi.fullrank_vi module blackjax.vi.meanfield_vi module blackjax.vi.multipathfinder module blackjax.vi.pathfinder module blackjax.vi.schrodinger_follmer module blackjax.vi.svgd module bracket_left (SliceInfo attribute) bracket_right (SliceInfo attribute) buffer (MetricRecipe attribute) buffer_idx (LowRankAdaptationState attribute) (LowRankMetricCoreState attribute) build_additive_step() (in module blackjax.mcmc.random_walk) build_coordinate_kernel() (in module blackjax.mcmc.slice) build_core() (MetricRecipe method) build_growing_window_schedule() (in module blackjax.adaptation.low_rank_adaptation) build_irmh() (in module blackjax.mcmc.random_walk) build_kernel (in module blackjax.mcmc.rmhmc) build_kernel() (in module blackjax.adaptation.laps_burn_in) (in module blackjax.mcmc.adjusted_mclmc) (in module blackjax.mcmc.adjusted_mclmc_dynamic) (in module blackjax.mcmc.barker) (in module blackjax.mcmc.dynamic_hmc) (in module blackjax.mcmc.elliptical_slice) (in module blackjax.mcmc.ghmc) (in module blackjax.mcmc.gist) (in module blackjax.mcmc.gist_step_size) (in module blackjax.mcmc.gist_trajectory_length) (in module blackjax.mcmc.hmc) (in module blackjax.mcmc.laplace_dynamic_hmc) (in module blackjax.mcmc.laplace_hmc) (in module blackjax.mcmc.mala) (in module blackjax.mcmc.marginal_latent_gaussian) (in module blackjax.mcmc.mclmc) (in module blackjax.mcmc.nuts) (in module blackjax.mcmc.periodic_orbital) (in module blackjax.mcmc.slice) (in module blackjax.ns.adaptive) (in module blackjax.ns.base) (in module blackjax.ns.from_mcmc) (in module blackjax.ns.nss) (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_persistent_sampling) (in module blackjax.smc.adaptive_tempered) (in module blackjax.smc.from_mcmc) (in module blackjax.smc.inner_kernel_tuning) (in module blackjax.smc.partial_posteriors_path) (in module blackjax.smc.persistent_sampling) (in module blackjax.smc.pretuning) (in module blackjax.smc.tempered) (in module blackjax.vi.svgd) build_meta_adaptation_core() (in module blackjax.adaptation.meta.builders) build_multi_chain_meta_core() (in module blackjax.adaptation.meta.builders) build_pretune() (in module blackjax.smc.pretuning) build_rmh() (in module blackjax.mcmc.random_walk) build_rmh_transition_energy() (in module blackjax.mcmc.random_walk) build_schedule() (in module blackjax.adaptation.staged_adaptation) build_swig_kernel() (in module blackjax.ns.nss) C C (Adaptation attribute) cgl_merge_two() (in module blackjax.adaptation.metric_buffers) cgl_update_batch() (in module blackjax.adaptation.metric_buffers) chees_adaptation() (in module blackjax.adaptation.chees_adaptation) CHEES_LENGTH_FLOOR_FACTOR (in module blackjax.adaptation.chees_adaptation) ChEESAdaptationState (class in blackjax.adaptation.chees_adaptation) chol_params (FRVIState attribute) compute_asymmetric_acceptance_ratio() (in module blackjax.mcmc.proposal) compute_log_persistent_weights() (in module blackjax.smc.persistent_sampling) compute_log_Z() (in module blackjax.smc.persistent_sampling) compute_num_live() (in module blackjax.ns.utils) compute_persistent_ess() (in module blackjax.smc.persistent_sampling) ConstrainedMCMCInfo (class in blackjax.ns.from_mcmc) ContourSGLDState (class in blackjax.sgmcmc.csgld) contract (Adaptation attribute), [1] contract_history() (in module blackjax.adaptation.laps_burn_in) control_variates() (in module blackjax.sgmcmc.gradients) coordinate_constrained_step() (in module blackjax.ns.nss) coordinate_proposal() (in module blackjax.mcmc.slice) (in module blackjax.ns.nss) coordinate_slice (in module blackjax) coordinate_slice() (in module blackjax.mcmc.slice) count (MomentBlock attribute) counts (AccumulatingSplitPopState attribute) covariance_proposal() (in module blackjax.ns.nss) csgld (in module blackjax) current_iteration (MEADSAdaptationState attribute) cutoff (MetricRecipe attribute) D da_state (ChEESAdaptationState attribute) data_mask (PartialPosteriorsSMCState attribute) default_measure_factory() (in module blackjax.smc.pretuning) default_metric() (in module blackjax.mcmc.metrics) delete_fn() (in module blackjax.ns.base) delta (MEADSAdaptationState attribute) dhmc (in module blackjax) diag_from_moment_block() (in module blackjax.adaptation.metric_buffers) diagnostics (MCLMCLRDAdaptationState attribute) dichotomy() (in module blackjax.smc.solver) direction_proposal() (in module blackjax.mcmc.slice) directions (PeriodicOrbitalState attribute) dmhmc (in module blackjax) doubling() (in module blackjax.mcmc.slice) draws_buffer (LowRankAdaptationState attribute) (LowRankMetricCoreState attribute) draws_singular_value_low_rank() (in module blackjax.adaptation.metric_estimators) dual_averaging_adaptation() (in module blackjax.adaptation.step_size) DualAveragingAdaptationState (class in blackjax.adaptation.step_size) dynamic_hmc (in module blackjax) 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) DynamicHMCState (class in blackjax.mcmc.dynamic_hmc) DynamicIntegrationState (class in blackjax.mcmc.trajectory) E eca_step() (in module blackjax.eca) EEVPD (AdaptationState attribute) EEVPD_wanted (AdaptationState attribute) effective_sample_size() (in module blackjax.diagnostics) eigenvalue_informativeness() (in module blackjax.adaptation.metric_estimators) elbo (FRVIInfo attribute) (MFVIInfo attribute) (PathfinderState attribute) EllipSliceInfo (class in blackjax.mcmc.elliptical_slice) EllipSliceState (class in blackjax.mcmc.elliptical_slice) elliptical_slice (in module blackjax) emits (MetricRecipe attribute) energy (GISTInfo attribute) (GISTStepSizeInfo attribute) (GISTTrajectoryLengthInfo attribute) (HMCInfo attribute) (LaplaceHMCInfo attribute) (NUTSInfo attribute) (Proposal attribute) energy_change (MCLMCInfo attribute) energy_idx (ContourSGLDState attribute) energy_pdf (ContourSGLDState attribute) ensemble_batch_buffer() (in module blackjax.adaptation.metric_buffers) ensemble_execute_fn() (in module blackjax.eca) EPS_FLOAT (in module blackjax.adaptation.chees_adaptation) epsadap_update (Adaptation attribute) equipartition_diagonal() (in module blackjax.adaptation.laps_burn_in) equipartition_diagonal_loss() (in module blackjax.adaptation.laps_burn_in) equipartition_fullrank() (in module blackjax.adaptation.laps_burn_in) equipartition_fullrank_loss() (in module blackjax.adaptation.laps_burn_in) esjd() (in module blackjax.smc.pretuning) ess() (in module blackjax.ns.utils) (in module blackjax.smc.ess) ess_bulk() (in module blackjax.diagnostics) ess_solver() (in module blackjax.smc.ess) ess_tail() (in module blackjax.diagnostics) estimator (MetricRecipe attribute) extend_params() (in module blackjax.smc.base) extract_meta_verdict() (in module blackjax.adaptation.meta.verdict) extract_multi_chain_verdict() (in module blackjax.adaptation.meta.verdict) F final (MetricCore attribute) finalise() (in module blackjax.ns.utils) find_reasonable_step_size() (in module blackjax.adaptation.step_size) fisher_block (FisherMassMatrixAdaptationState attribute) fisher_score_diagonal() (in module blackjax.adaptation.metric_estimators) fisher_score_diagonal_from_moments() (in module blackjax.adaptation.metric_estimators) fisher_score_low_rank() (in module blackjax.adaptation.metric_estimators) FisherMassMatrixAdaptationState (class in blackjax.adaptation.mass_matrix) fixed_order() (in module blackjax.mcmc.slice) FRVIInfo (class in blackjax.vi.fullrank_vi) FRVIState (class in blackjax.vi.fullrank_vi) fullrank_vi (in module blackjax) G gamma (MetricRecipe attribute) (PathfinderState attribute) gaussian_euclidean() (in module blackjax.mcmc.metrics) gaussian_euclidean_low_rank() (in module blackjax.mcmc.metrics) gaussian_riemannian() (in module blackjax.mcmc.metrics) generate_fullrank_logdensity() (in module blackjax.vi.fullrank_vi) generate_meanfield_logdensity() (in module blackjax.vi.meanfield_vi) get_diag_reference (MetricBuffer attribute) get_filter_adapt_info_fn() (in module blackjax.adaptation.base) get_first_row() (in module blackjax.ns.utils) get_moments (MetricBuffer attribute) get_support (MetricBuffer attribute) get_theta_star (LaplaceMarginal attribute) ghmc (in module blackjax) GHMCState (class in blackjax.mcmc.ghmc) gist_step_size (in module blackjax) gist_trajectory_length (in module blackjax) GISTInfo (class in blackjax.mcmc.gist) GISTState (class in blackjax.mcmc.gist) GISTStepSizeInfo (class in blackjax.mcmc.gist_step_size) GISTTrajectoryLengthInfo (class in blackjax.mcmc.gist_trajectory_length) grad_estimator() (in module blackjax.sgmcmc.gradients) grad_position (PathfinderState attribute) grads_buffer (LowRankAdaptationState attribute) (LowRankMetricCoreState attribute) H halton_sequence() (in module blackjax.mcmc.dynamic_hmc) handle_nans() (in module blackjax.adaptation.adjusted_mclmc_adaptation) (in module blackjax.adaptation.mclmc_adaptation) history (AdaptationState attribute) History (class in blackjax.adaptation.laps_burn_in) hmc (in module blackjax) 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 (StagedAdaptationState attribute) implicit_midpoint() (in module blackjax.mcmc.integrators) info (AdaptationInfo attribute) (ConstrainedMCMCInfo attribute), [1] init (in module blackjax.mcmc.gist_step_size) (in module blackjax.mcmc.gist_trajectory_length) (in module blackjax.mcmc.nuts) (in module blackjax.mcmc.rmhmc) (in module blackjax.smc.adaptive_persistent_sampling) (in module blackjax.smc.adaptive_tempered) (MetricBuffer attribute) (MetricCore attribute) init() (in module blackjax.mcmc.adjusted_mclmc) (in module blackjax.mcmc.adjusted_mclmc_dynamic) (in module blackjax.mcmc.barker) (in module blackjax.mcmc.dynamic_hmc) (in module blackjax.mcmc.elliptical_slice) (in module blackjax.mcmc.ghmc) (in module blackjax.mcmc.gist) (in module blackjax.mcmc.hmc) (in module blackjax.mcmc.laplace_dynamic_hmc) (in module blackjax.mcmc.laplace_hmc) (in module blackjax.mcmc.mala) (in module blackjax.mcmc.marginal_latent_gaussian) (in module blackjax.mcmc.mclmc) (in module blackjax.mcmc.periodic_orbital) (in module blackjax.mcmc.slice) (in module blackjax.ns.adaptive) (in module blackjax.ns.base) (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.inner_kernel_tuning) (in module blackjax.smc.partial_posteriors_path) (in module blackjax.smc.persistent_sampling) (in module blackjax.smc.pretuning) (in module blackjax.smc.tempered) (in module blackjax.vi.schrodinger_follmer) (in module blackjax.vi.svgd) init_integrator() (in module blackjax.ns.integrator) initial_state (Adaptation attribute), [1] initialize() (in module blackjax.adaptation.laps_burn_in) inner (LateStartState attribute) inner_kernel_tuning (in module blackjax) inverse_mass_matrix (AdaptationState attribute) (FisherMassMatrixAdaptationState attribute) (LowRankMetricCoreState attribute) (MassMatrixAdaptationState attribute) (MCLMCAdaptationState attribute) (MCLMCLRDAdaptationState attribute) (PathfinderAdaptationState attribute) (StagedAdaptationState attribute) inverse_mass_matrix_from_particles() (in module blackjax.smc.tuning.from_particles) irmh (in module blackjax) irmh_as_top_level_api() (in module blackjax.mcmc.random_walk) is_accepted (BarkerInfo attribute) (ConstrainedMCMCInfo attribute), [1] (GISTInfo attribute) (GISTStepSizeInfo attribute) (GISTTrajectoryLengthInfo attribute) (HMCInfo attribute) (LaplaceHMCInfo attribute) (MALAInfo attribute) (MarginalInfo attribute) (RWInfo attribute) (SliceInfo attribute) is_divergent (GISTInfo attribute) (GISTStepSizeInfo attribute) (GISTTrajectoryLengthInfo attribute) (HMCInfo attribute) (LaplaceHMCInfo attribute) (NUTSInfo attribute) is_no_return_rejected (GISTTrajectoryLengthInfo attribute) is_turning (NUTSInfo attribute) isokinetic_mclachlan (in module blackjax.mcmc.integrators) isokinetic_omelyan (in module blackjax.mcmc.integrators) isokinetic_velocity_verlet (in module blackjax.mcmc.integrators) isokinetic_yoshida (in module blackjax.mcmc.integrators) iteration (AdaptationState attribute) (PersistentSMCState attribute) iterative_uturn_numpyro() (in module blackjax.mcmc.termination) IterativeUTurnState (class in blackjax.mcmc.termination) K kinetic_change (MCLMCInfo attribute) L L (AdaptationState attribute) (MCLMCAdaptationState attribute) (MCLMCLRDAdaptationState attribute) lam (LowRankAdaptationState attribute) (LowRankInverseMassMatrix attribute), [1] laplace_dhmc (in module blackjax) laplace_dmhmc (in module blackjax) laplace_hmc (in module blackjax) laplace_lbfgs_grad_evals() (in module blackjax.mcmc.laplace_marginal) laplace_marginal_factory() (in module blackjax.mcmc.laplace_marginal) laplace_mhmc (in module blackjax) LaplaceDynamicHMCState (class in blackjax.mcmc.laplace_dynamic_hmc) LaplaceHMCInfo (class in blackjax.mcmc.laplace_marginal) LaplaceHMCState (class in blackjax.mcmc.laplace_hmc) LaplaceMarginal (class in blackjax.mcmc.laplace_marginal) laps() (in module blackjax.adaptation.laps) late_start() (in module blackjax.adaptation.metric_buffers) LateStartState (class in blackjax.adaptation.metric_buffers) lbfgs_converged (LaplaceHMCInfo attribute) lbfgs_error (LaplaceHMCInfo attribute) lbfgs_hit_maxiter (LaplaceHMCInfo attribute) lbfgs_inverse_hessian_to_low_rank_metric() (in module blackjax.mcmc.metrics) lbfgs_iter_num (LaplaceHMCInfo attribute) leftmost_state (Trajectory attribute) live_covariance() (in module blackjax.ns.nss) live_covariance_factor() (in module blackjax.ns.nss) live_widths() (in module blackjax.ns.nss) log1mexp() (in module blackjax.ns.utils) log_ess() (in module blackjax.smc.ess) log_likelihood_increment (SMCInfo attribute) log_marginal (LaplaceMarginal attribute) log_step_size (DualAveragingAdaptationState attribute) log_step_size_avg (DualAveragingAdaptationState attribute) log_step_size_moving_average (ChEESAdaptationState attribute) log_trajectory_length_moving_average (ChEESAdaptationState attribute) LOG_UPDATE_CLIP (in module blackjax.adaptation.chees_adaptation) log_weights() (in module blackjax.ns.utils) log_Z (PersistentSMCState property) logdensities (PeriodicOrbitalState attribute) logdensities_grad (PeriodicOrbitalState attribute) logdensity (BarkerState attribute) (DynamicHMCState attribute) (EllipSliceState attribute) (GHMCState attribute) (GISTState attribute) (HMCState attribute) (LaplaceDynamicHMCState attribute) (LaplaceHMCState attribute) (MALAState attribute) (MarginalState attribute) (MCLMCInfo attribute) (RWState attribute) (SliceState attribute) logdensity_estimator() (in module blackjax.sgmcmc.gradients) logdensity_grad (BarkerState attribute) (DynamicHMCState attribute) (GHMCState attribute) (GISTState attribute) (HMCState attribute) (LaplaceDynamicHMCState attribute) (LaplaceHMCState attribute) (MALAState attribute) (MarginalState attribute) logp (MultipathfinderState attribute) logq (MultipathfinderState attribute) logX (NSIntegrator attribute), [1] logX() (in module blackjax.ns.utils) logZ (NSIntegrator attribute), [1] logZ_live (NSIntegrator attribute), [1] lookup_recipe() (in module blackjax.adaptation.metric_recipes) LowRankAdaptationState (class in blackjax.adaptation.low_rank_adaptation) LowRankInverseMassMatrix (class in blackjax.mcmc.metrics) LowRankMetricCoreState (class in blackjax.adaptation.metric_recipes) Lratio_lowerbound (in module blackjax.adaptation.adjusted_mclmc_adaptation) Lratio_upperbound (in module blackjax.adaptation.adjusted_mclmc_adaptation) M m2 (MomentBlock attribute) (WelfordAlgorithmState attribute) m2s (AccumulatingSplitPopState attribute) main() (in module blackjax.progress_reader) make_adaptation_L() (in module blackjax.adaptation.mclmc_adaptation) make_L_step_size_adaptation() (in module blackjax.adaptation.mclmc_adaptation) mala (in module blackjax) MALAInfo (class in blackjax.mcmc.mala) MALAState (class in blackjax.mcmc.mala) map_fn() (in module blackjax.smc.base) map_kernel() (in module blackjax.smc.base) 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) max_rank (MetricRecipe attribute) maximum_eigenvalue() (in module blackjax.adaptation.meads_adaptation) mclachlan (in module blackjax.mcmc.integrators) mclmc (in module blackjax) mclmc_find_L_and_step_size() (in module blackjax.adaptation.mclmc_adaptation) mclmc_lrd_warmup() (in module blackjax.adaptation.mclmc_lrd_adaptation) MCLMCAdaptationState (class in blackjax.adaptation.mclmc_adaptation) MCLMCInfo (class in blackjax.mcmc.mclmc) MCLMCLRDAdaptationState (class in blackjax.adaptation.mclmc_lrd_adaptation) meads_adaptation() (in module blackjax.adaptation.meads_adaptation) MEADSAdaptationState (class in blackjax.adaptation.meads_adaptation) mean (MomentBlock attribute) (WelfordAlgorithmState attribute) meanfield_vi (in module blackjax) means (AccumulatingSplitPopState attribute) merge_block_ring() (in module blackjax.adaptation.metric_buffers) merge_trajectories() (in module blackjax.mcmc.trajectory) MetricBuffer (class in blackjax.adaptation.metric_buffers) MetricCore (class in blackjax.adaptation.metric_recipes) MetricRecipe (class in blackjax.adaptation.metric_recipes) MFVIInfo (class in blackjax.vi.meanfield_vi) MFVIState (class in blackjax.vi.meanfield_vi) mgrad_gaussian (in module blackjax) mhmc (in module blackjax) module blackjax blackjax.adaptation blackjax.adaptation.adjusted_mclmc_adaptation blackjax.adaptation.base blackjax.adaptation.chees_adaptation blackjax.adaptation.laps blackjax.adaptation.laps_burn_in blackjax.adaptation.low_rank_adaptation blackjax.adaptation.mass_matrix blackjax.adaptation.mclmc_adaptation blackjax.adaptation.mclmc_lrd_adaptation blackjax.adaptation.meads_adaptation blackjax.adaptation.meta blackjax.adaptation.meta.builders blackjax.adaptation.meta.verdict blackjax.adaptation.meta_adaptation blackjax.adaptation.metric_buffers blackjax.adaptation.metric_estimators blackjax.adaptation.metric_recipes blackjax.adaptation.pathfinder_adaptation blackjax.adaptation.staged_adaptation blackjax.adaptation.step_size blackjax.adaptation.window_adaptation blackjax.diagnostics blackjax.eca blackjax.mcmc blackjax.mcmc.adjusted_mclmc blackjax.mcmc.adjusted_mclmc_dynamic blackjax.mcmc.barker blackjax.mcmc.diffusions blackjax.mcmc.dynamic_hmc blackjax.mcmc.elliptical_slice blackjax.mcmc.ghmc blackjax.mcmc.gist blackjax.mcmc.gist_step_size blackjax.mcmc.gist_trajectory_length blackjax.mcmc.hmc blackjax.mcmc.integrators blackjax.mcmc.laplace_dynamic_hmc blackjax.mcmc.laplace_hmc blackjax.mcmc.laplace_marginal blackjax.mcmc.mala blackjax.mcmc.marginal_latent_gaussian blackjax.mcmc.mclmc blackjax.mcmc.metrics blackjax.mcmc.nuts blackjax.mcmc.periodic_orbital blackjax.mcmc.proposal blackjax.mcmc.random_walk blackjax.mcmc.rmhmc blackjax.mcmc.slice blackjax.mcmc.termination blackjax.mcmc.trajectory blackjax.ns blackjax.ns.adaptive blackjax.ns.base blackjax.ns.from_mcmc blackjax.ns.integrator blackjax.ns.nss blackjax.ns.utils blackjax.progress_reader 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_persistent_sampling blackjax.smc.adaptive_tempered blackjax.smc.base blackjax.smc.ess blackjax.smc.from_mcmc blackjax.smc.inner_kernel_tuning blackjax.smc.partial_posteriors_path blackjax.smc.persistent_sampling blackjax.smc.pretuning blackjax.smc.resampling blackjax.smc.solver blackjax.smc.tempered blackjax.smc.tuning blackjax.smc.tuning.from_kernel_info blackjax.smc.tuning.from_particles blackjax.smc.waste_free blackjax.vi blackjax.vi.fullrank_vi blackjax.vi.meanfield_vi blackjax.vi.multipathfinder blackjax.vi.pathfinder blackjax.vi.schrodinger_follmer blackjax.vi.svgd MomentBlock (class in blackjax.adaptation.metric_buffers) momentum (EllipSliceInfo attribute) (GHMCState attribute) (GISTInfo attribute) (GISTStepSizeInfo attribute) (GISTTrajectoryLengthInfo attribute) (HMCInfo attribute) (IterativeUTurnState attribute) (LaplaceHMCInfo attribute) (NUTSInfo attribute) (SGNHTState attribute) momentum_sum (IterativeUTurnState attribute) (Trajectory attribute) mu (DualAveragingAdaptationState attribute) (FRVIState attribute) (MFVIState attribute) mu_star (LowRankAdaptationState attribute) (LowRankMetricCoreState attribute) multi_approximate() (in module blackjax.vi.multipathfinder) multinomial() (in module blackjax.smc.resampling) multinomial_hmc (in module blackjax) multinomial_hmc_proposal() (in module blackjax.mcmc.hmc) multipathfinder (in module blackjax) MultipathfinderState (class in blackjax.vi.multipathfinder) N nan_reject() (in module blackjax.adaptation.laps_burn_in) ndims (Adaptation attribute) needs (MetricRecipe attribute) no_nans() (in module blackjax.adaptation.laps_burn_in) nonans (MCLMCInfo attribute) nonreversible_slice_sampling() (in module blackjax.mcmc.proposal) norm_factor (Adaptation attribute) normal() (in module blackjax.mcmc.random_walk) normal_random_walk() (in module blackjax.mcmc.random_walk) NSInfo (class in blackjax.ns.base) NSIntegrator (class in blackjax.ns.integrator) NSState (class in blackjax.ns.base) num_adaptation_samples (Adaptation attribute) num_expansions (SliceInfo attribute) num_integration_steps (GISTInfo attribute) (GISTStepSizeInfo attribute) (GISTTrajectoryLengthInfo attribute) (HMCInfo attribute) (LaplaceHMCInfo attribute) (NUTSInfo attribute) num_particles (PersistentSMCState property) num_shrink (SliceInfo attribute) num_skipped (LateStartState attribute) num_states (Trajectory attribute) num_steps_to_uturn() (in module blackjax.mcmc.gist_trajectory_length) num_steps_to_uturn_forward (GISTTrajectoryLengthInfo attribute) num_steps_to_uturn_reverse (GISTTrajectoryLengthInfo attribute) num_trajectory_expansions (NUTSInfo attribute) nuts (in module blackjax) NUTSInfo (class in blackjax.mcmc.nuts) O observables (Adaptation attribute), [1] (History attribute) observables_for_bias (Adaptation attribute), [1] omelyan (in module blackjax.mcmc.integrators) opt_state (FRVIState attribute) (MFVIState attribute) optim_state (ChEESAdaptationState attribute) OPTIMAL_TARGET_ACCEPTANCE_RATE (in module blackjax.adaptation.chees_adaptation) orbital_hmc (in module blackjax) overdamped_langevin() (in module blackjax.mcmc.diffusions) (in module blackjax.sgmcmc.diffusions) P parameter_override (SMCInfoWithParameterDistribution attribute) (StateWithParameterOverride attribute) parameters (AdaptationResults attribute) pareto_khat() (in module blackjax.diagnostics) partial_posteriors_smc (in module blackjax) PartialPosteriorsSMCState (class in blackjax.smc.partial_posteriors_path) particles (NSInfo attribute), [1] (NSState attribute), [1] (PartialPosteriorsSMCState attribute) (PersistentSMCState property) (SMCState attribute) (TemperedSMCState attribute) particles_covariance_matrix() (in module blackjax.smc.tuning.from_particles) particles_means() (in module blackjax.smc.tuning.from_particles) particles_stds() (in module blackjax.smc.tuning.from_particles) path_states (MultipathfinderState attribute) pathfinder (in module blackjax) 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) persistent_log_likelihoods (PersistentSMCState attribute) persistent_log_Z (PersistentSMCState attribute) persistent_particles (PersistentSMCState attribute) persistent_sampling_smc (in module blackjax) persistent_weights (PersistentSMCState property) PersistentSMCState (class in blackjax.smc.persistent_sampling) PersistentStateInfo (class in blackjax.smc.persistent_sampling) position (BarkerState attribute) (ContourSGLDState attribute) (DynamicHMCState attribute) (EllipSliceState attribute) (GHMCState attribute) (GISTState attribute) (HMCState attribute) (LaplaceDynamicHMCState attribute) (LaplaceHMCState attribute) (MALAState attribute) (MarginalState attribute) (PathfinderState attribute) (RWState attribute) (SchrodingerFollmerState attribute) (SGNHTState attribute) (SliceState attribute) position_sigma (MEADSAdaptationState attribute) positions (PeriodicOrbitalState attribute) potential_scale_reduction() (in module blackjax.diagnostics) pretuning (in module blackjax) progressive_biased_sampling() (in module blackjax.mcmc.proposal) progressive_uniform_sampling() (in module blackjax.mcmc.proposal) proposal (BarkerInfo attribute) Proposal (class in blackjax.mcmc.proposal) proposal (DynamicExpansionState attribute) (DynamicIntegrationState attribute) (HMCInfo attribute) (LaplaceHMCInfo attribute) (MarginalInfo attribute) (RWInfo attribute) proposal_generator() (in module blackjax.mcmc.proposal) provenance (MetricRecipe attribute) provides (MetricRecipe attribute) psis_weights() (in module blackjax.diagnostics) (in module blackjax.vi.multipathfinder) push_split (MetricBuffer attribute) R r_end (Adaptation attribute) random_generator_arg (ChEESAdaptationState attribute) (DynamicHMCState attribute) (LaplaceDynamicHMCState attribute) random_order() (in module blackjax.mcmc.slice) rbf_kernel() (in module blackjax.vi.svgd) read_progress() (in module blackjax.progress_reader) recompute_counter (LowRankAdaptationState attribute) (LowRankMetricCoreState attribute) REGISTRY (in module blackjax.adaptation.metric_recipes) reject_constrained_step() (in module blackjax.ns.from_mcmc) remove_padding() (in module blackjax.smc.persistent_sampling) reorder_trajectories() (in module blackjax.mcmc.trajectory) representation (MetricRecipe attribute) resample_from_persistent() (in module blackjax.smc.persistent_sampling) reset_window_buffer() (in module blackjax.adaptation.metric_buffers) residual() (in module blackjax.smc.resampling) return_all_adapt_info() (in module blackjax.adaptation.base) reverse_step_index (GISTStepSizeInfo attribute) rhat() (in module blackjax.diagnostics) rho (MFVIState attribute) rightmost_state (Trajectory attribute) rmh (in module blackjax) rmh_as_top_level_api() (in module blackjax.mcmc.random_walk) rmh_proposal() (in module blackjax.mcmc.random_walk) rmhmc (in module blackjax) run_eca() (in module blackjax.eca) RWInfo (class in blackjax.mcmc.random_walk) RWState (class in blackjax.mcmc.random_walk) S safe_energy_diff() (in module blackjax.mcmc.proposal) sample() (in module blackjax.ns.utils) (in module blackjax.vi.fullrank_vi) (in module blackjax.vi.meanfield_vi) (in module blackjax.vi.pathfinder) (in module blackjax.vi.schrodinger_follmer) sample_covariance_eigh_low_rank() (in module blackjax.adaptation.metric_estimators) sample_direction() (in module blackjax.mcmc.slice) sample_size (WelfordAlgorithmState attribute) sample_theta (LaplaceMarginal attribute) sample_variance_diagonal() (in module blackjax.adaptation.metric_estimators) sampler_state (StateWithParameterOverride attribute) samples (MultipathfinderState attribute) save_num (Adaptation attribute) schrodinger_follmer (in module blackjax) SchrodingerFollmerState (class in blackjax.vi.schrodinger_follmer) search_exhausted (GISTStepSizeInfo attribute) seed_low_rank_sigma_from_grad() (in module blackjax.adaptation.metric_recipes) select_top_eigenvalues_by_informativeness() (in module blackjax.adaptation.metric_estimators) sghmc (in module blackjax) sghmc() (in module blackjax.sgmcmc.diffusions) sgld (in module blackjax) sgnht (in module blackjax) sgnht() (in module blackjax.sgmcmc.diffusions) SGNHTState (class in blackjax.sgmcmc.sgnht) sigma (LowRankAdaptationState attribute) (LowRankInverseMassMatrix attribute), [1] slice (GHMCState attribute) slice_constrained_step() (in module blackjax.ns.nss) slice_sampling (in module blackjax) SliceInfo (class in blackjax.mcmc.slice) SliceState (class in blackjax.mcmc.slice) smc_info (SMCInfoWithParameterDistribution attribute) SMCInfo (class in blackjax.smc.base) SMCInfoWithParameterDistribution (class in blackjax.smc.pretuning) SMCState (class in blackjax.smc.base) solve_theta (LaplaceMarginal attribute) solve_theta_with_info (LaplaceMarginal attribute) ss_state (LowRankAdaptationState attribute) (PathfinderAdaptationState attribute) (StagedAdaptationState attribute) staged_adaptation() (in module blackjax.adaptation.staged_adaptation) StagedAdaptationState (class in blackjax.adaptation.staged_adaptation) state (AdaptationInfo attribute) (AdaptationResults attribute) (Proposal attribute) StateWithParameterOverride (class in blackjax.smc.inner_kernel_tuning) static_binomial_sampling() (in module blackjax.mcmc.proposal) static_integration() (in module blackjax.mcmc.trajectory) static_progressive_integration() (in module blackjax.mcmc.trajectory) step (ChEESAdaptationState attribute) (DualAveragingAdaptationState attribute) (DynamicExpansionState attribute) (DynamicIntegrationState attribute) step() (in module blackjax.smc.base) (in module blackjax.smc.persistent_sampling) (in module blackjax.vi.fullrank_vi) (in module blackjax.vi.meanfield_vi) (in module blackjax.vi.schrodinger_follmer) step_count (AdaptationState attribute) step_index (GISTStepSizeInfo attribute) (StepSizeTuningParameter attribute) step_size (AdaptationState attribute), [1] (ChEESAdaptationState attribute) (GISTStepSizeInfo attribute) (LowRankAdaptationState attribute) (MCLMCAdaptationState attribute) (MCLMCLRDAdaptationState attribute) (MEADSAdaptationState attribute) (PathfinderAdaptationState attribute) (StagedAdaptationState attribute) step_size_selector() (in module blackjax.mcmc.gist_step_size) stepping_out() (in module blackjax.mcmc.slice) steps_per_sample (AdaptationState attribute) stepsize_adaptation_state (AdaptationState attribute) StepSizeTuningParameter (class in blackjax.mcmc.gist_step_size) stopping (History attribute) stratified() (in module blackjax.smc.resampling) subiter (EllipSliceInfo attribute) sum_log_p_accept (Proposal attribute) summary_statistics_fn() (Adaptation method), [1] support_gate (MetricRecipe attribute) svgd (in module blackjax) swig_as_top_level_api() (in module blackjax.ns.nss) systematic() (in module blackjax.smc.resampling) T tempered_smc (in module blackjax) TemperedSMCState (class in blackjax.smc.tempered) tempering_param (PersistentSMCState property) (TemperedSMCState attribute) tempering_schedule (PersistentSMCState attribute) termination_state (DynamicExpansionState attribute) (DynamicIntegrationState attribute) theta (EllipSliceInfo attribute) theta_star (LaplaceDynamicHMCState attribute) (LaplaceHMCState attribute) time (SchrodingerFollmerState attribute) Trajectory (class in blackjax.mcmc.trajectory) trajectory (DynamicExpansionState attribute) (DynamicIntegrationState attribute) trajectory_leftmost_state (NUTSInfo attribute) trajectory_length (ChEESAdaptationState attribute) trajectory_rightmost_state (NUTSInfo attribute) TrajectoryState (in module blackjax.mcmc.proposal) tuning_parameter (GISTInfo attribute) (GISTStepSizeInfo attribute) (GISTTrajectoryLengthInfo attribute) U U (LowRankAdaptationState attribute) (LowRankInverseMassMatrix attribute), [1] U_grad_x (MarginalState attribute) U_x (MarginalState attribute) uniform_prior() (in module blackjax.ns.utils) unshared_parameters_and_step_fn() (in module blackjax.smc.from_mcmc) update (MetricBuffer attribute) (MetricCore attribute) update() (Adaptation method), [1] update_and_take_last() (in module blackjax.smc.base) update_history() (in module blackjax.adaptation.laps_burn_in) update_history_scalar() (in module blackjax.adaptation.laps_burn_in) update_info (NSInfo attribute), [1] (PersistentStateInfo attribute) (SMCInfo attribute) update_integrator() (in module blackjax.ns.integrator) update_median_heuristic() (in module blackjax.vi.svgd) update_parameter_distribution() (in module blackjax.smc.pretuning) update_parameters (SMCState attribute) update_scale_from_acceptance_rate() (in module blackjax.smc.tuning.from_kernel_info) update_waste_free() (in module blackjax.smc.waste_free) update_with_mcmc_take_last() (in module blackjax.ns.from_mcmc) V velocity_verlet (in module blackjax.mcmc.integrators) W waste_free_smc() (in module blackjax.smc.waste_free) wc_state (MassMatrixAdaptationState attribute) weight (Proposal attribute) weighted_empirical_mean() (in module blackjax.adaptation.chees_adaptation) weights (History attribute) (PartialPosteriorsSMCState attribute) (PeriodicOrbitalState attribute) (SMCState attribute) (TemperedSMCState attribute) welford_algorithm() (in module blackjax.adaptation.mass_matrix) welford_dense() (in module blackjax.adaptation.metric_estimators) welford_diagonal() (in module blackjax.adaptation.metric_estimators) WelfordAlgorithmState (class in blackjax.adaptation.mass_matrix) while_cond() (Adaptation method) while_steps_num() (in module blackjax.adaptation.laps) while_with_info() (in module blackjax.eca) window_adaptation() (in module blackjax.adaptation.window_adaptation) window_adaptation_low_rank() (in module blackjax.adaptation.low_rank_adaptation) WindowAdaptationState (in module blackjax.adaptation.window_adaptation) with_isokinetic_maruyama() (in module blackjax.mcmc.integrators) write_pos (AccumulatingSplitPopState attribute) X xi (SGNHTState attribute) Y yoshida (in module blackjax.mcmc.integrators)
