Index A | B | C | D | E | F | G | H | I | K | L | M | N | P | R | S | T | U | V | W | X | Z A affine_transform (aspire.transforms.CompositeTransform attribute) (aspire.transforms.FlowPreconditioningTransform attribute) AffineTransform (class in aspire.transforms) args (aspire.utils.CallHistory attribute), [1] array_to_namespace() (aspire.samples.BaseSamples method) asarray() (in module aspire.utils) aspire module Aspire (class in aspire) (class in aspire.aspire) aspire.aspire module aspire.flows module aspire.flows.base module aspire.flows.jax module aspire.flows.jax.flows module aspire.flows.jax.utils module aspire.flows.torch module aspire.flows.torch.flows module aspire.history module aspire.plot module aspire.samplers module aspire.samplers.base module aspire.samplers.importance module aspire.samplers.mcmc module aspire.samplers.smc module aspire.samplers.smc.base module aspire.samplers.smc.blackjax module aspire.samplers.smc.emcee module aspire.samplers.smc.minipcn module aspire.samples module aspire.transforms module aspire.utils module aspire_instance (aspire.utils.PoolHandler property) AspireFile (class in aspire.utils) at_temperature() (aspire.samples.PTMCMCSamples method) auto_checkpoint() (aspire.Aspire method) (aspire.aspire.Aspire method) autocorrelation_time (aspire.samples.MCMCSamples attribute) B backend_str (aspire.samplers.base.Sampler attribute) BaseSamples (class in aspire.samples) BaseTorchFlow (class in aspire.flows.torch.flows) BaseTransform (class in aspire.transforms) beta (aspire.history.SMCHistory attribute) (aspire.samples.SMCSamples attribute) betas (aspire.samples.PTMCMCSamples attribute) BetaScheduleError BlackJAXSMC (class in aspire.samplers.smc.blackjax) bounded_to_unbounded (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.transforms.CompositeTransform attribute) (aspire.transforms.FlowPreconditioningTransform attribute) bounded_transform (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.transforms.CompositeTransform attribute) (aspire.transforms.FlowPreconditioningTransform attribute) BoundedTransform (class in aspire.transforms) build_checkpoint_state() (aspire.samplers.base.Sampler method) (aspire.samplers.smc.base.SMCSampler method) burn_in (aspire.samples.MCMCSamples attribute) C CallHistory (class in aspire.utils) chain (aspire.samples.MCMCSamples property) chain_checkpoint_path (aspire.samplers.mcmc.MCMCSampler attribute) chain_dataset_name (aspire.samplers.mcmc.MCMCSampler attribute) chain_shape (aspire.samples.MCMCSamples attribute) checkpoint_mcmc_chain() (aspire.samplers.mcmc.MCMCSampler method) close_pool (aspire.utils.PoolHandler attribute) cold_chain() (aspire.samples.PTMCMCSamples method) CompositeTransform (class in aspire.transforms) compute_weights() (aspire.Samples method) (aspire.samples.Samples method) concatenate() (aspire.samples.BaseSamples class method) config_dict() (aspire.Aspire method) (aspire.aspire.Aspire method) (aspire.flows.base.Flow method) (aspire.samplers.base.Sampler method) (aspire.samplers.smc.base.SMCSampler method) (aspire.transforms.AffineTransform method) (aspire.transforms.BaseTransform method) (aspire.transforms.BoundedTransform method) (aspire.transforms.CompositeTransform method) (aspire.transforms.FlowTransform method) (aspire.transforms.LogitTransform method) (aspire.transforms.PeriodicTransform method) (aspire.transforms.ProbitTransform method) configure_logger() (in module aspire.utils) convert_dtype() (in module aspire.utils) convert_to_samples() (aspire.Aspire method) (aspire.aspire.Aspire method) copy_array() (in module aspire.utils) current_target_efficiency() (aspire.samplers.smc.base.SMCSampler method) D data_transform (aspire.flows.base.Flow attribute) decode_dtype() (in module aspire.utils) decode_from_hdf5() (in module aspire.utils) decode_samples() (in module aspire.utils) DEFAULT_BETA_TOLERANCE (in module aspire.samplers.smc.base) default_checkpoint_callback() (aspire.samplers.base.Sampler method) default_file_checkpoint_callback() (aspire.samplers.base.Sampler method) default_mcmc_chain_file_checkpoint_callback() (aspire.samplers.mcmc.MCMCSampler method) determine_backend_name() (in module aspire.utils) determine_beta() (aspire.samplers.smc.base.SMCSampler method) device (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.flows.base.Flow attribute) (aspire.samples.BaseSamples attribute) (aspire.transforms.CompositeTransform attribute) (aspire.transforms.FlowPreconditioningTransform attribute) dims (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.flows.base.Flow attribute) (aspire.samplers.base.Sampler attribute) (aspire.samples.BaseSamples property) disable_gradients() (in module aspire.utils) draw_initial_samples() (aspire.samplers.mcmc.MCMCSampler method) dtype (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.flows.jax.flows.FlowJax attribute) (aspire.flows.torch.flows.BaseTorchFlow attribute) (aspire.samplers.base.Sampler attribute) (aspire.samples.BaseSamples attribute) (aspire.transforms.BaseTransform attribute) dump_pickle_to_hdf() (in module aspire.utils) dump_state() (in module aspire.utils) E eff_target (aspire.history.SMCHistory attribute) effective_sample_size (aspire.Samples attribute) (aspire.samples.Samples attribute) effective_sample_size() (in module aspire.utils) efficiency (aspire.Samples property) (aspire.samples.Samples property) Emcee (class in aspire.samplers.mcmc) EmceeSMC (class in aspire.samplers.smc.emcee) enable_pool() (aspire.Aspire method) (aspire.aspire.Aspire method) enable_scipy_array_api() (in module aspire.utils) encode_dtype() (in module aspire.utils) encode_for_hdf5() (in module aspire.utils) encode_samples() (in module aspire.utils) eps (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.transforms.CompositeTransform attribute) (aspire.transforms.FlowPreconditioningTransform attribute) (aspire.transforms.LogitTransform attribute) (aspire.transforms.ProbitTransform attribute) ess (aspire.history.SMCHistory attribute) ess_target (aspire.history.SMCHistory attribute) eta (aspire.flows.torch.flows.ZukoFlowMatching attribute) evidence (aspire.Samples attribute) (aspire.samples.Samples attribute) evidence_error (aspire.Samples attribute) (aspire.samples.Samples attribute) F fit() (aspire.Aspire method) (aspire.aspire.Aspire method) (aspire.flows.base.Flow method) (aspire.flows.jax.flows.FlowJax method) (aspire.flows.torch.flows.BaseTorchFlow method) (aspire.flows.torch.flows.ZukoFlow method) (aspire.transforms.AffineTransform method) (aspire.transforms.BaseTransform method) (aspire.transforms.BoundedTransform method) (aspire.transforms.CompositeTransform method) (aspire.transforms.FlowPreconditioningTransform method) (aspire.transforms.IdentityTransform method) (aspire.transforms.LogitTransform method) (aspire.transforms.PeriodicTransform method) (aspire.transforms.ProbitTransform method) fit_data_transform() (aspire.flows.base.Flow method) fit_kwargs (aspire.transforms.FlowPreconditioningTransform attribute) fit_preconditioning_transform() (aspire.samplers.base.Sampler method) flow (aspire.Aspire property) (aspire.aspire.Aspire property) (aspire.flows.torch.flows.BaseTorchFlow property) (aspire.flows.torch.flows.ZukoFlow attribute) (aspire.transforms.FlowPreconditioningTransform attribute) Flow (class in aspire.flows.base) flow_backend (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.transforms.FlowPreconditioningTransform attribute) flow_kwargs (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.transforms.FlowPreconditioningTransform attribute) flow_matching (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.transforms.FlowPreconditioningTransform attribute) FlowHistory (class in aspire.history) FlowJax (class in aspire.flows.jax.flows) FlowPreconditioningTransform (class in aspire.transforms) FlowTransform (class in aspire.transforms) forward() (aspire.flows.jax.flows.FlowJax method) (aspire.flows.torch.flows.ZukoFlow method) (aspire.transforms.AffineTransform method) (aspire.transforms.BaseTransform method) (aspire.transforms.BoundedTransform method) (aspire.transforms.CompositeTransform method) (aspire.transforms.FlowPreconditioningTransform method) (aspire.transforms.IdentityTransform method) (aspire.transforms.LogitTransform method) (aspire.transforms.PeriodicTransform method) (aspire.transforms.ProbitTransform method) from_chain() (aspire.samples.MCMCSamples class method) (aspire.samples.PTMCMCSamples class method) from_dict() (aspire.samples.BaseSamples class method) from_samples() (aspire.samples.BaseSamples class method) from_unit_interval() (aspire.transforms.BoundedTransform method) function_id() (in module aspire.utils) G get_bijection_class() (in module aspire.flows.jax.utils) get_flow() (in module aspire.flows.jax) (in module aspire.flows.jax.utils) get_flow_function_class() (in module aspire.flows.jax.utils) get_flow_wrapper() (in module aspire.flows) get_package_version() (in module aspire.utils) get_sampler_class() (aspire.Aspire method) (aspire.aspire.Aspire method) H history (aspire.samplers.base.Sampler attribute) History (class in aspire.history) I IdentityTransform (class in aspire.transforms) ImportanceSampler (class in aspire.samplers.importance) infer_device() (in module aspire.utils) init_flow() (aspire.Aspire method) (aspire.aspire.Aspire method) init_sampler() (aspire.Aspire method) (aspire.aspire.Aspire method) interval_check() (aspire.transforms.BoundedTransform method) inverse() (aspire.flows.jax.flows.FlowJax method) (aspire.flows.torch.flows.ZukoFlow method) (aspire.transforms.AffineTransform method) (aspire.transforms.BaseTransform method) (aspire.transforms.BoundedTransform method) (aspire.transforms.CompositeTransform method) (aspire.transforms.FlowPreconditioningTransform method) (aspire.transforms.IdentityTransform method) (aspire.transforms.LogitTransform method) (aspire.transforms.PeriodicTransform method) (aspire.transforms.ProbitTransform method) inverse_rescale() (aspire.flows.base.Flow method) IS_NAMESPACE_FUNCTIONS (in module aspire.utils) K key (aspire.flows.jax.flows.FlowJax attribute) (aspire.samplers.smc.blackjax.BlackJAXSMC attribute) kwargs (aspire.utils.CallHistory attribute), [1] L last_checkpoint_bytes (aspire.samplers.base.Sampler property) last_checkpoint_state (aspire.samplers.base.Sampler property) load() (aspire.flows.base.Flow class method) (aspire.flows.jax.flows.FlowJax class method) (aspire.flows.torch.flows.BaseTorchFlow class method) (aspire.history.History class method) (aspire.history.SMCHistory class method) (aspire.samples.BaseSamples class method) (aspire.transforms.BaseTransform class method) load_checkpoint_from_file() (aspire.samplers.base.Sampler method) load_flow() (aspire.Aspire method) (aspire.aspire.Aspire method) load_from_h5_file() (in module aspire.utils) loc (aspire.flows.jax.flows.FlowJax attribute) (aspire.flows.torch.flows.BaseTorchFlow attribute) log_evidence (aspire.Samples attribute) (aspire.samples.Samples attribute) (aspire.samples.SMCSamples attribute) log_evidence_error (aspire.Samples attribute) (aspire.samples.Samples attribute) (aspire.samples.SMCSamples attribute) log_evidence_ratio() (aspire.samples.SMCSamples method) log_evidence_ratio_variance() (aspire.samples.SMCSamples method) log_evidence_stepping_stone() (aspire.samples.PTMCMCSamples method) log_evidence_thermodynamic_integration() (aspire.samples.PTMCMCSamples method) log_likelihood (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.samples.BaseSamples attribute) log_likelihood() (aspire.samplers.base.Sampler method) log_likelihood_wrapper() (aspire.samplers.mcmc.ParallelTemperedMCMCSampler method) log_norm_ratio (aspire.history.SMCHistory attribute) log_norm_ratio_var (aspire.history.SMCHistory attribute) log_p_t() (aspire.samples.SMCSamples method) log_prior (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.samples.BaseSamples attribute) log_prior() (aspire.samplers.base.Sampler method) log_prior_wrapper() (aspire.samplers.mcmc.ParallelTemperedMCMCSampler method) log_prob() (aspire.flows.base.Flow method) (aspire.flows.jax.flows.FlowJax method) (aspire.flows.torch.flows.ZukoFlow method) (aspire.samplers.mcmc.MCMCSampler method) (aspire.samplers.mcmc.NumpyMCMCSampler method) (aspire.samplers.smc.base.NumpySMCSampler method) (aspire.samplers.smc.base.SMCSampler method) (aspire.samplers.smc.blackjax.BlackJAXSMC method) (aspire.samplers.smc.minipcn.MiniPCNSMC method) log_q (aspire.samples.BaseSamples attribute) log_w (aspire.Samples attribute) (aspire.samples.Samples attribute) log_weights() (aspire.samples.SMCSamples method) logger (in module aspire.aspire) (in module aspire.flows) (in module aspire.flows.base) (in module aspire.flows.jax.flows) (in module aspire.flows.torch.flows) (in module aspire.samplers.base) (in module aspire.samplers.mcmc) (in module aspire.samplers.smc.base) (in module aspire.samplers.smc.blackjax) (in module aspire.samplers.smc.emcee) (in module aspire.samples) (in module aspire.transforms) (in module aspire.utils) logit() (in module aspire.utils) LogitTransform (class in aspire.transforms) logsumexp() (in module aspire.utils) loss_fn() (aspire.flows.torch.flows.ZukoFlow method) (aspire.flows.torch.flows.ZukoFlowMatching method) lower (aspire.transforms.BoundedTransform attribute) (aspire.transforms.PeriodicTransform attribute) M mcmc_acceptance (aspire.history.SMCHistory attribute) mcmc_autocorr (aspire.history.SMCHistory attribute) MCMCSampler (class in aspire.samplers.mcmc) MCMCSamples (class in aspire.samples) minimum_chain_ndim (aspire.samples.MCMCSamples attribute) (aspire.samples.PTMCMCSamples attribute) MiniPCN (class in aspire.samplers.mcmc) MiniPCNSMC (class in aspire.samplers.smc.minipcn) module aspire aspire.aspire aspire.flows aspire.flows.base aspire.flows.jax aspire.flows.jax.flows aspire.flows.jax.utils aspire.flows.torch aspire.flows.torch.flows aspire.history aspire.plot aspire.samplers aspire.samplers.base aspire.samplers.importance aspire.samplers.mcmc aspire.samplers.smc aspire.samplers.smc.base aspire.samplers.smc.blackjax aspire.samplers.smc.emcee aspire.samplers.smc.minipcn aspire.samples aspire.transforms aspire.utils mutate() (aspire.samplers.smc.base.SMCSampler method) (aspire.samplers.smc.blackjax.BlackJAXSMC method) (aspire.samplers.smc.emcee.EmceeSMC method) (aspire.samplers.smc.minipcn.MiniPCNSMC method) N n_chains (aspire.samples.MCMCSamples property) n_likelihood_evaluations (aspire.Aspire property) (aspire.aspire.Aspire property) (aspire.samplers.base.Sampler attribute) n_steps (aspire.samples.MCMCSamples property) n_temps (aspire.samples.PTMCMCSamples property) name (aspire.transforms.AffineTransform attribute) (aspire.transforms.BoundedTransform attribute) (aspire.transforms.LogitTransform attribute) (aspire.transforms.PeriodicTransform attribute) (aspire.transforms.ProbitTransform attribute) new_instance() (aspire.transforms.CompositeTransform method) (aspire.transforms.FlowPreconditioningTransform method) (aspire.transforms.FlowTransform method) NumpyMCMCSampler (class in aspire.samplers.mcmc) NumpySMCSampler (class in aspire.samplers.smc.base) P parallelize_prior (aspire.utils.PoolHandler attribute) ParallelTemperedMCMCSampler (class in aspire.samplers.mcmc) parameters (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.samplers.base.Sampler attribute) (aspire.samples.BaseSamples attribute) (aspire.transforms.CompositeTransform attribute) (aspire.transforms.FlowPreconditioningTransform attribute) periodic_parameters (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.transforms.CompositeTransform attribute) (aspire.transforms.FlowPreconditioningTransform attribute) PeriodicTransform (class in aspire.transforms) plot() (aspire.history.SMCHistory method) plot_beta() (aspire.history.SMCHistory method) plot_chain() (aspire.samples.PTMCMCSamples method) plot_comparison() (in module aspire.plot) plot_corner() (aspire.Samples method) (aspire.samples.BaseSamples method) (aspire.samples.Samples method) plot_eff_target() (aspire.history.SMCHistory method) plot_ess() (aspire.history.SMCHistory method) plot_ess_target() (aspire.history.SMCHistory method) plot_history_comparison() (in module aspire.plot) plot_log_norm_ratio() (aspire.history.SMCHistory method) plot_loss() (aspire.history.FlowHistory method) plot_mcmc_acceptance() (aspire.history.SMCHistory method) plot_mcmc_autocorr() (aspire.history.SMCHistory method) plot_quantile_bands() (aspire.history.SMCHistory method) plot_sample_history() (aspire.history.SMCHistory method) pool (aspire.utils.PoolHandler attribute) PoolHandler (class in aspire.utils) post_process() (aspire.samples.MCMCSamples method) prior_bounds (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.transforms.FlowPreconditioningTransform attribute) prior_flow (aspire.samplers.base.Sampler attribute) ProbitTransform (class in aspire.transforms) PTMCMCSamples (class in aspire.samples) R recursively_save_to_h5_file() (in module aspire.utils) rejection_sample() (aspire.Samples method) (aspire.samples.Samples method) requires_prior_bounds (aspire.transforms.AffineTransform attribute) (aspire.transforms.BoundedTransform attribute) (aspire.transforms.LogitTransform attribute) (aspire.transforms.PeriodicTransform attribute) (aspire.transforms.ProbitTransform attribute) resample() (aspire.samples.SMCSamples method) rescale() (aspire.flows.base.Flow method) resolve_dtype() (in module aspire.utils) resolve_xp() (in module aspire.utils) restore_from_checkpoint() (aspire.samplers.base.Sampler method) (aspire.samplers.smc.base.SMCSampler method) resume_from_file() (aspire.Aspire class method) (aspire.aspire.Aspire class method) rng (aspire.samplers.mcmc.MCMCSampler attribute) (aspire.samplers.smc.base.SMCSampler attribute) (aspire.samplers.smc.blackjax.BlackJAXSMC attribute) (aspire.samplers.smc.minipcn.MiniPCNSMC attribute) S safe_to_device() (in module aspire.utils) sample() (aspire.flows.base.Flow method) (aspire.flows.jax.flows.FlowJax method) (aspire.flows.torch.flows.ZukoFlow method) (aspire.samplers.base.Sampler method) (aspire.samplers.importance.ImportanceSampler method) (aspire.samplers.mcmc.Emcee method) (aspire.samplers.mcmc.MiniPCN method) (aspire.samplers.smc.base.SMCSampler method) (aspire.samplers.smc.blackjax.BlackJAXSMC method) (aspire.samplers.smc.emcee.EmceeSMC method) (aspire.samplers.smc.minipcn.MiniPCNSMC method) sample_and_log_prob() (aspire.flows.base.Flow method) (aspire.flows.jax.flows.FlowJax method) (aspire.flows.torch.flows.ZukoFlow method) sample_flow() (aspire.Aspire method) (aspire.aspire.Aspire method) sample_history (aspire.history.SMCHistory attribute) sample_posterior() (aspire.Aspire method) (aspire.aspire.Aspire method) sampler (aspire.Aspire property) (aspire.aspire.Aspire property) Sampler (class in aspire.samplers.base) Samples (class in aspire) (class in aspire.samples) save() (aspire.flows.base.Flow method) (aspire.flows.jax.flows.FlowJax method) (aspire.flows.torch.flows.BaseTorchFlow method) (aspire.history.FlowHistory method) (aspire.history.History method) (aspire.history.SMCHistory method) (aspire.samples.BaseSamples method) (aspire.transforms.BaseTransform method) (aspire.transforms.FlowPreconditioningTransform method) save_checkpoint_to_hdf() (aspire.samplers.base.Sampler method) save_config() (aspire.Aspire method) (aspire.aspire.Aspire method) save_config_to_json() (aspire.Aspire method) (aspire.aspire.Aspire method) save_flow() (aspire.Aspire method) (aspire.aspire.Aspire method) save_sampler_config() (aspire.Aspire method) (aspire.aspire.Aspire method) scale (aspire.flows.jax.flows.FlowJax attribute) (aspire.flows.torch.flows.BaseTorchFlow attribute) scaled_weights (aspire.Samples property) (aspire.samples.Samples property) serialize_checkpoint() (aspire.samplers.base.Sampler method) sigmoid() (in module aspire.utils) SMCHistory (class in aspire.history) SMCSampler (class in aspire.samplers.smc.base) SMCSamples (class in aspire.samples) subsample() (aspire.samples.PTMCMCSamples method) T target_efficiency (aspire.samplers.smc.base.SMCSampler property) thin (aspire.samples.MCMCSamples attribute) to_dataframe() (aspire.Samples method) (aspire.samples.BaseSamples method) (aspire.samples.Samples method) to_dict() (aspire.samples.BaseSamples method) (aspire.utils.CallHistory method) to_namespace() (aspire.Samples method) (aspire.samples.BaseSamples method) (aspire.samples.Samples method) to_numpy() (aspire.Samples method) (aspire.samples.BaseSamples method) (aspire.samples.MCMCSamples method) (aspire.samples.Samples method) (aspire.samples.SMCSamples method) (in module aspire.utils) to_standard_samples() (aspire.samples.SMCSamples method) to_unit_interval() (aspire.transforms.BoundedTransform method) track_calls() (in module aspire.utils) training_loss (aspire.history.FlowHistory attribute) U unnormalized_log_weights() (aspire.samples.SMCSamples method) update_at_indices() (in module aspire.utils) upper (aspire.transforms.BoundedTransform attribute) (aspire.transforms.PeriodicTransform attribute) V validation_loss (aspire.history.FlowHistory attribute) W weights (aspire.Samples attribute) (aspire.samples.Samples attribute) X x (aspire.samples.BaseSamples attribute) xp (aspire.Aspire attribute) (aspire.aspire.Aspire attribute) (aspire.flows.base.Flow attribute) (aspire.flows.jax.flows.FlowJax attribute) (aspire.flows.torch.flows.BaseTorchFlow attribute) (aspire.samplers.base.Sampler attribute) (aspire.samples.BaseSamples attribute) (aspire.transforms.BaseTransform attribute) Z ZukoFlow (class in aspire.flows.torch.flows) ZukoFlowMatching (class in aspire.flows.torch.flows)