Index

_

__class_vars__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __private_attributes__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_complete__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_computed_fields__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_core_schema__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_custom_init__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_decorators__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_extra__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_fields__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute)

__pydantic_fields_set__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_generic_metadata__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_parent_namespace__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_post_init__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_private__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_root_model__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_serializer__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __pydantic_validator__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) __signature__ (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute)

A

adapt_mass_matrix (gwkokab.analysis.core.inference_io.NumpyroNUTSSamplerConfig attribute) adapt_step_size (gwkokab.analysis.core.inference_io.NumpyroNUTSSamplerConfig attribute) add_rule() (gwkokab.parameters.RelationMesh method) all_constraint (in module gwkokab.models.constraints) alternate_datasets (gwkokab.analysis.core.inference_io.DiscretePELoader attribute) alternate_distance_priors (gwkokab.analysis.core.inference_io.DiscretePELoader attribute) alternate_mass_priors (gwkokab.analysis.core.inference_io.DiscretePELoader attribute) alternate_spin_priors (gwkokab.analysis.core.inference_io.DiscretePELoader attribute) alternate_waveforms (gwkokab.analysis.core.inference_io.AnalyticalPELoader attribute) analysis_base_arg_parser() (in module gwkokab.analysis.core.analysis_base)

analysis_name (gwkokab.analysis.core.analysis_base.AnalysisBase attribute) AnalysisBase (class in gwkokab.analysis.core.analysis_base) analytical_arg_parser() (in module gwkokab.analysis.core.analytical_base) analytical_poisson_likelihood_fn() (in module gwkokab.inference) (in module gwkokab.inference.poissonlikelihood_utils) AnalyticalBase (class in gwkokab.analysis.core.analytical_base) AnalyticalPELoader (class in gwkokab.analysis.core.inference_io) any_constraint (in module gwkokab.models.constraints) args_constraints (gwkokab.models.mass.GenericSmoothedPowerlawMassRatio attribute) autocorr_1d() (in module gwkokab_scripts.pe_diagnostics)

B

banana_error() (in module gwkokab.errors) batch_and_remainder() (in module gwkokab.utils.tools) batch_size (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) bce_logits_loss_fn() (in module gwkokab.utils.train) beta_dist_concentrations_to_mean_variance() (in module gwkokab.utils) (in module gwkokab.utils.math) beta_dist_mean_variance_to_concentrations() (in module gwkokab.utils) (in module gwkokab.utils.math)

BetaFromMeanVar() (in module gwkokab.models) (in module gwkokab.models.spin) BlockTransform (class in gwkokab.models.transformations) BrokenPowerlaw (class in gwkokab.models) (class in gwkokab.models.mass) BrokenPowerlawTwoPeak (class in gwkokab.models) (class in gwkokab.models.mass)

C

calculate_dist_layouts() (in module gwkokab.analysis.utils.marginals) calculate_js_metrics() (in module gwkokab.analysis.core.synthetic_pe) calculate_marginals_over_axes() (in module gwkokab.analysis.utils.marginals) cart_to_polar() (in module gwkokab.utils) (in module gwkokab.utils.transformations) cart_to_spherical() (in module gwkokab.utils) (in module gwkokab.utils.transformations) cdf() (gwkokab.models.utils.DoublyTruncatedPowerLaw method) (gwkokab.models.utils.ScaledMixture method) chain_batch_size (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) CHAIN_GROUP_FORMAT (in module gwkokab.analysis.utils.literals) chain_method (gwkokab.analysis.core.inference_io.NumpyroMCMCConfig attribute) check() (gwkokab.analysis.core.utils.SampleTransformer method) check_leaks (gwkokab.analysis.core.analysis_base.AnalysisBase attribute) check_min_concentration_for_beta_dist() (in module gwkokab.analysis.utils.checks) CHI_1 (gwkokab.parameters.Parameters attribute) CHI_2 (gwkokab.parameters.Parameters attribute) chi_costilt_to_chiz() (in module gwkokab.utils) (in module gwkokab.utils.transformations) CHI_MINUS (gwkokab.parameters.Parameters attribute) chi_p_from_components() (in module gwkokab.utils) (in module gwkokab.utils.transformations) chieff() (in module gwkokab.utils) (in module gwkokab.utils.transformations) CHIRP_MASS (gwkokab.parameters.Parameters attribute) chirp_mass() (in module gwkokab.utils) (in module gwkokab.utils.transformations) CHIRP_MASS_DETECTOR (gwkokab.parameters.Parameters attribute) chirp_mass_from_m1_q() (in module gwkokab.utils.transformations) CHIRP_MASS_SOURCE (gwkokab.parameters.Parameters attribute) classify_model_parameters() (gwkokab.analysis.core.analysis_base.AnalysisBase method) codomain (gwkokab.models.transformations.BlockTransform attribute) (gwkokab.models.transformations.ComponentMassesAndRedshiftToDetectedMassAndRedshift attribute) (gwkokab.models.transformations.ComponentMassesToChirpMassAndDelta attribute) (gwkokab.models.transformations.ComponentMassesToChirpMassAndSymmetricMassRatio attribute) (gwkokab.models.transformations.ComponentMassesToMassRatioAndSecondaryMass attribute) (gwkokab.models.transformations.ComponentMassesToPrimaryMassAndMassRatio attribute) (gwkokab.models.transformations.ComponentMassesToTotalMassAndMassRatio attribute) (gwkokab.models.transformations.DeltaToSymmetricMassRatio attribute) (gwkokab.models.transformations.PrimaryMassAndMassRatioToComponentMassesTransform attribute) (gwkokab.models.transformations.RedshiftToLuminosityDistance attribute) (gwkokab.models.transformations.SourceMassAndRedshiftToDetectedMassAndRedshift attribute)

color (gwkokab.analysis.utils.marginals.PlotStyle attribute) component_cdf() (gwkokab.models.utils.ScaledMixture method) component_distributions (gwkokab.models.utils.ScaledMixture property) component_log_probs() (gwkokab.models.utils.ScaledMixture method) component_mean (gwkokab.models.utils.ScaledMixture property) component_sample() (gwkokab.models.utils.ScaledMixture method) component_variance (gwkokab.models.utils.ScaledMixture property) ComponentMassesAndRedshiftToDetectedMassAndRedshift (class in gwkokab.models.transformations) ComponentMassesToChirpMassAndDelta (class in gwkokab.models.transformations) ComponentMassesToChirpMassAndSymmetricMassRatio (class in gwkokab.models.transformations) ComponentMassesToMassRatioAndSecondaryMass (class in gwkokab.models.transformations) ComponentMassesToPrimaryMassAndMassRatio (class in gwkokab.models.transformations) ComponentMassesToTotalMassAndMassRatio (class in gwkokab.models.transformations) compute_batched_marginals() (in module gwkokab.analysis.utils.marginals) compute_ess() (in module gwkokab_scripts.pe_diagnostics) compute_rhat() (in module gwkokab_scripts.pe_diagnostics) console (in module gwkokab_scripts.flowMC_info) coords (gwkokab.analysis.core.synthetic_pe.SyntheticAnalyticalPE attribute) (gwkokab.analysis.core.synthetic_pe.SyntheticDiscretePE attribute) COS_IOTA (gwkokab.parameters.Parameters attribute) COS_TILT_1 (gwkokab.parameters.Parameters attribute) COS_TILT_2 (gwkokab.parameters.Parameters attribute) Cosmology (class in gwkokab.cosmology) cosmology (gwkokab.models.transformations.RedshiftToLuminosityDistance attribute) create_grid() (in module gwkokab_scripts.param_lens) cumtrapz() (in module gwkokab.utils.math) custom_format() (in module gwkokab.analysis.utils.logger)

D

data_loader (gwkokab.analysis.core.analytical_base.AnalyticalBase attribute) (gwkokab.analysis.core.discrete_base.DiscreteBase attribute) dataset (gwkokab.analysis.core.synthetic_pe.SyntheticDiscretePE attribute) Dc (gwkokab.cosmology.Cosmology property) dDcdz() (gwkokab.cosmology.Cosmology method) dDLdz() (gwkokab.cosmology.Cosmology method) debug_nans (gwkokab.analysis.core.analysis_base.AnalysisBase attribute) decreasing_vector (in module gwkokab.models.constraints) default_cosmology() (in module gwkokab.cosmology) default_datasets (gwkokab.analysis.core.inference_io.DiscretePELoader attribute) default_distance_prior (gwkokab.analysis.core.inference_io.DiscretePELoader attribute) default_mass_prior (gwkokab.analysis.core.inference_io.DiscretePELoader attribute) default_relation_mesh() (in module gwkokab.parameters) default_spin_prior (gwkokab.analysis.core.inference_io.DiscretePELoader attribute) default_waveform (gwkokab.analysis.core.inference_io.AnalyticalPELoader attribute) DELTA_M (gwkokab.parameters.Parameters attribute) delta_m() (in module gwkokab.utils) (in module gwkokab.utils.transformations) delta_m_to_symmetric_mass_ratio() (in module gwkokab.utils) (in module gwkokab.utils.transformations) DeltaToSymmetricMassRatio (class in gwkokab.models.transformations) dense_mass (gwkokab.analysis.core.inference_io.NumpyroNUTSSamplerConfig attribute) dependencies (gwkokab.models.utils.LazyJointDistribution attribute) derive_only() (gwkokab.parameters.RelationMesh method) derive_parameters (gwkokab.analysis.core.synthetic_events.SyntheticEventsBase attribute) (gwkokab.analysis.core.synthetic_pe.SyntheticDiscretePE attribute) deserialize() (gwkokab.analysis.core.flowMC_base.Sampler method) DETECTION_TIME (gwkokab.parameters.Parameters attribute) diagnose_parameter() (in module gwkokab_scripts.pe_diagnostics) diagnostic_label() (in module gwkokab_scripts.pe_diagnostics) DirichletElement() (in module gwkokab.analysis.utils.priors)

discrete_arg_parser() (in module gwkokab.analysis.core.discrete_base) discrete_poisson_likelihood_fn() (in module gwkokab.inference) (in module gwkokab.inference.poissonlikelihood_utils) discrete_waveform (gwkokab.analysis.core.synthetic_pe.SyntheticAnalyticalPE attribute) DiscreteBase (class in gwkokab.analysis.core.discrete_base) DiscretePELoader (class in gwkokab.analysis.core.inference_io) display_config_table() (in module gwkokab_scripts.flowMC_info) display_diagnostics_panel() (in module gwkokab_scripts.flowMC_info) display_loop_table() (in module gwkokab_scripts.flowMC_info) display_suggestions() (in module gwkokab_scripts.flowMC_info) DL (gwkokab.cosmology.Cosmology property) DL_to_z() (gwkokab.cosmology.Cosmology method) domain (gwkokab.models.transformations.BlockTransform attribute) (gwkokab.models.transformations.ComponentMassesAndRedshiftToDetectedMassAndRedshift attribute) (gwkokab.models.transformations.ComponentMassesToChirpMassAndDelta attribute) (gwkokab.models.transformations.ComponentMassesToChirpMassAndSymmetricMassRatio attribute) (gwkokab.models.transformations.ComponentMassesToMassRatioAndSecondaryMass attribute) (gwkokab.models.transformations.ComponentMassesToPrimaryMassAndMassRatio attribute) (gwkokab.models.transformations.ComponentMassesToTotalMassAndMassRatio attribute) (gwkokab.models.transformations.DeltaToSymmetricMassRatio attribute) (gwkokab.models.transformations.PrimaryMassAndMassRatioToComponentMassesTransform attribute) (gwkokab.models.transformations.RedshiftToLuminosityDistance attribute) (gwkokab.models.transformations.SourceMassAndRedshiftToDetectedMassAndRedshift attribute) doubly_truncated_power_law_cdf() (in module gwkokab.models.utils) doubly_truncated_power_law_icdf() (in module gwkokab.models.utils) doubly_truncated_power_law_log_norm_constant() (in module gwkokab.models.utils) doubly_truncated_power_law_log_prob() (in module gwkokab.models.utils) DoublyTruncatedPowerLaw (class in gwkokab.models.utils) driver() (gwkokab.analysis.core.analysis_base.AnalysisBase method) (gwkokab.analysis.core.flowMC_base.FlowMCBase method) (gwkokab.analysis.core.numpyro_base.NumpyroBase method) dVcdz() (gwkokab.cosmology.Cosmology method)

E

ECCENTRICITY (gwkokab.parameters.Parameters attribute) EccentricityMattersCore (class in gwkokab.analysis.ecc_matters.common) EccentricityMattersFAnalyticalAnalysis (class in gwkokab.analysis.ecc_matters.analytical) EccentricityMattersFDiscreteAnalysis (class in gwkokab.analysis.ecc_matters.discrete) EccentricityMattersModel() (in module gwkokab.analysis.ecc_matters.common) EccentricityMattersNAnalyticalAnalysis (class in gwkokab.analysis.ecc_matters.analytical) EccentricityMattersNDiscreteAnalysis (class in gwkokab.analysis.ecc_matters.discrete) EFFECTIVE_SPIN (gwkokab.parameters.Parameters attribute) eq() (gwkokab.models.transformations.BlockTransform method) (gwkokab.models.transformations.ComponentMassesAndRedshiftToDetectedMassAndRedshift method) (gwkokab.models.transformations.ComponentMassesToChirpMassAndDelta method) (gwkokab.models.transformations.ComponentMassesToChirpMassAndSymmetricMassRatio method) (gwkokab.models.transformations.ComponentMassesToMassRatioAndSecondaryMass method) (gwkokab.models.transformations.ComponentMassesToPrimaryMassAndMassRatio method) (gwkokab.models.transformations.ComponentMassesToTotalMassAndMassRatio method) (gwkokab.models.transformations.DeltaToSymmetricMassRatio method) (gwkokab.models.transformations.PrimaryMassAndMassRatioToComponentMassesTransform method) (gwkokab.models.transformations.SourceMassAndRedshiftToDetectedMassAndRedshift method)

error_function_registry (gwkokab.analysis.core.synthetic_pe.SyntheticDiscretePE property) error_params_filename (gwkokab.analysis.core.synthetic_pe.SyntheticDiscretePE attribute) ErrorFunctionRegistryType (in module gwkokab.analysis.core.synthetic_pe) eta_from_q() (in module gwkokab.utils) (in module gwkokab.utils.transformations) event_paths (gwkokab.analysis.core.inference_io.AnalyticalPELoader attribute) event_slices (gwkokab.models.transformations.BlockTransform attribute) expand_arguments() (in module gwkokab.analysis.utils.common) ExtendedSupportTransformedDistribution (class in gwkokab.models.utils)

F

filename (gwkokab.analysis.core.synthetic_events.SyntheticEventsBase attribute) (gwkokab.analysis.core.synthetic_pe.SyntheticAnalyticalPE attribute) (gwkokab.analysis.core.synthetic_pe.SyntheticDiscretePE attribute) filenames (gwkokab.analysis.core.inference_io.DiscretePELoader attribute) fill_between_kwargs (gwkokab.analysis.utils.marginals.PlotStyle attribute) find_heuristic_step_size (gwkokab.analysis.core.inference_io.NumpyroNUTSSamplerConfig attribute) flowMC_analytical_poisson_likelihood() (in module gwkokab.inference) (in module gwkokab.inference.flowMC_analytical_poisson_likelihood) flowMC_arg_parser (in module gwkokab.analysis.core.flowMC_base) flowMC_discrete_poisson_likelihood() (in module gwkokab.inference) (in module gwkokab.inference.flowMC_discrete_poisson_likelihood)

FlowMCBase (class in gwkokab.analysis.core.flowMC_base) FlowMCGlobalConfig (class in gwkokab.analysis.core.inference_io) forward_mode_differentiation (gwkokab.analysis.core.inference_io.NumpyroNUTSSamplerConfig attribute) from_inverse_transform_sampling() (gwkokab.analysis.core.synthetic_events.SyntheticEventsBase method) from_json() (gwkokab.analysis.core.inference_io.AnalyticalPELoader class method) (gwkokab.analysis.core.inference_io.DiscretePELoader class method) (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig class method) (gwkokab.analysis.core.inference_io.NumpyroGlobalConfig class method) (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader class method) (gwkokab.analysis.core.inference_io.SamplerConfig static method) from_structured() (in module gwkokab.analysis.core.utils)

G

GaussianPrimaryMassRatio (class in gwkokab.models.mass) GaussianSpinModel() (in module gwkokab.models) (in module gwkokab.models.spin) generate_heuristics() (in module gwkokab_scripts.flowMC_info) generate_marginal_probs() (in module gwkokab.analysis.utils.marginals) generate_parameter_estimates() (gwkokab.analysis.core.synthetic_pe.SyntheticAnalyticalPE method) (gwkokab.analysis.core.synthetic_pe.SyntheticDiscretePE method) generate_report module generate_report() (in module generate_report) GenericSmoothedPowerlawMassRatio (class in gwkokab.models.mass) GenericTiltModel() (in module gwkokab.models) (in module gwkokab.models.spin) get_estimators() (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader method) get_likelihood_fn() (in module gwkokab.inference.factory) get_processed_priors() (in module gwkokab.analysis.utils.priors) global_thinning (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) gwkokab module gwkokab.analysis module gwkokab.analysis.core module gwkokab.analysis.core.analysis_base module gwkokab.analysis.core.analytical_base module gwkokab.analysis.core.discrete_base module gwkokab.analysis.core.flowMC_base module gwkokab.analysis.core.inference_io module gwkokab.analysis.core.numpyro_base module gwkokab.analysis.core.synthetic_events module gwkokab.analysis.core.synthetic_pe module gwkokab.analysis.core.utils module gwkokab.analysis.ecc_matters module gwkokab.analysis.ecc_matters.analytical module gwkokab.analysis.ecc_matters.common module gwkokab.analysis.ecc_matters.discrete module gwkokab.analysis.ecc_matters.synthetic_events module gwkokab.analysis.multisource module gwkokab.analysis.multisource.analytical module gwkokab.analysis.multisource.common module gwkokab.analysis.multisource.discrete module gwkokab.analysis.n_pls_m_gs module gwkokab.analysis.n_pls_m_gs.analytical module gwkokab.analysis.n_pls_m_gs.common module gwkokab.analysis.n_pls_m_gs.discrete module gwkokab.analysis.n_pls_m_gs.synthetic_events module gwkokab.analysis.subpopulation module gwkokab.analysis.subpopulation.analytical module gwkokab.analysis.subpopulation.common module gwkokab.analysis.subpopulation.discrete module gwkokab.analysis.utils module gwkokab.analysis.utils.checks module gwkokab.analysis.utils.common module gwkokab.analysis.utils.jenks module gwkokab.analysis.utils.literals module gwkokab.analysis.utils.logger module gwkokab.analysis.utils.marginals module gwkokab.analysis.utils.priors module gwkokab.analysis.utils.regex module gwkokab.constants module gwkokab.cosmology module

gwkokab.errors module gwkokab.inference module gwkokab.inference.factory module gwkokab.inference.flowMC_analytical_poisson_likelihood module gwkokab.inference.flowMC_discrete_poisson_likelihood module gwkokab.inference.numpyro_analytical_poisson_likelihood module gwkokab.inference.numpyro_discrete_poisson_likelihood module gwkokab.inference.poissonlikelihood_utils module gwkokab.models module gwkokab.models.constraints module gwkokab.models.hybrids module gwkokab.models.mass module gwkokab.models.redshift module gwkokab.models.spin module gwkokab.models.sundry module gwkokab.models.transformations module gwkokab.models.utils module gwkokab.models.wrappers module gwkokab.parameters module gwkokab.poisson_mean module gwkokab.utils module gwkokab.utils.exceptions module gwkokab.utils.kernel module gwkokab.utils.math module gwkokab.utils.path module gwkokab.utils.tools module gwkokab.utils.train module gwkokab.utils.transformations module gwkokab.version module gwkokab_scripts module gwkokab_scripts.batch_scatter2d module gwkokab_scripts.batch_scatter3d module gwkokab_scripts.chain_plot module gwkokab_scripts.corner_overplot module gwkokab_scripts.corner_plot module gwkokab_scripts.diag_mass_matrix module gwkokab_scripts.ess_evolution_plot module gwkokab_scripts.ess_plot module gwkokab_scripts.flowMC_info module gwkokab_scripts.h5repack module gwkokab_scripts.hist_overplot module gwkokab_scripts.joint_plot module gwkokab_scripts.param_lens module gwkokab_scripts.pe_diagnostics module gwkokab_scripts.r_hat_plot module gwkokab_scripts.scatter2d module gwkokab_scripts.scatter3d module gwkokab_scripts.trace_plot module GWTC4EffectiveSpinSkewNormalModel (class in gwkokab.models) (class in gwkokab.models.spin)

H

histogram_pdf() (in module gwkokab.analysis.core.synthetic_pe)

history_window (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) Ho (gwkokab.cosmology.Cosmology attribute)

I

icdf() (gwkokab.models.utils.DoublyTruncatedPowerLaw method) IdentitySampleTransformer (class in gwkokab.analysis.core.utils) increasing_vector (in module gwkokab.models.constraints) infer_n_dims() (in module gwkokab_scripts.flowMC_info) INFERENCE_OUTPUT_FILENAME (in module gwkokab.analysis.utils.literals)

init_rng_seed() (gwkokab.analysis.core.utils.PRNGKeyMixin class method) injection_generator_parser() (in module gwkokab.analysis.core.synthetic_events) inverse_mass_matrix (gwkokab.analysis.core.inference_io.NumpyroNUTSSamplerConfig attribute) is_delta_error (gwkokab.analysis.core.synthetic_pe.SyntheticDiscretePE attribute) is_discrete (gwkokab.models.utils.ScaledMixture property)

J

jit_model_args (gwkokab.analysis.core.inference_io.NumpyroMCMCConfig attribute)

JointDistribution (class in gwkokab.models.utils) js_divergence_from_histograms() (in module gwkokab.analysis.core.synthetic_pe)

K

kernel (gwkokab.analysis.core.inference_io.NumpyroGlobalConfig attribute)

L

label (gwkokab.analysis.utils.marginals.PlotStyle attribute) LazyJointDistribution (class in gwkokab.models.utils) learning_rate (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) likelihood_fn (gwkokab.analysis.core.analytical_base.AnalyticalBase attribute) (gwkokab.analysis.core.discrete_base.DiscreteBase attribute) line_plot_kwargs (gwkokab.analysis.utils.marginals.PlotStyle attribute) load() (gwkokab.analysis.core.inference_io.AnalyticalPELoader method) (gwkokab.analysis.core.inference_io.DiscretePELoader method) load_config() (in module gwkokab_scripts.flowMC_info) load_file() (gwkokab.analysis.core.inference_io.AnalyticalPELoader class method) (gwkokab.analysis.core.inference_io.DiscretePELoader class method) load_model() (in module gwkokab.utils) (in module gwkokab.utils.train) load_samples() (in module gwkokab_scripts.pe_diagnostics) loader (gwkokab.analysis.core.inference_io.PoissonMeanEstimationLoader attribute) Local_Global_Sampler_Bundle (class in gwkokab.analysis.core.flowMC_base) local_sampler_name (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) local_thinning (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) log_abs_det_jacobian() (gwkokab.analysis.core.utils.IdentitySampleTransformer method) (gwkokab.analysis.core.utils.SampleTransformer method) (gwkokab.models.transformations.BlockTransform method) (gwkokab.models.transformations.ComponentMassesAndRedshiftToDetectedMassAndRedshift method) (gwkokab.models.transformations.ComponentMassesToChirpMassAndDelta method) (gwkokab.models.transformations.ComponentMassesToChirpMassAndSymmetricMassRatio method) (gwkokab.models.transformations.ComponentMassesToMassRatioAndSecondaryMass method) (gwkokab.models.transformations.ComponentMassesToPrimaryMassAndMassRatio method) (gwkokab.models.transformations.ComponentMassesToTotalMassAndMassRatio method) (gwkokab.models.transformations.DeltaToSymmetricMassRatio method) (gwkokab.models.transformations.PrimaryMassAndMassRatioToComponentMassesTransform method) (gwkokab.models.transformations.RedshiftToLuminosityDistance method) (gwkokab.models.transformations.SourceMassAndRedshiftToDetectedMassAndRedshift method) log_chirp_mass() (in module gwkokab.utils) (in module gwkokab.utils.transformations) log_device_info() (in module gwkokab.analysis.utils.logger) log_gwkokab_info() (in module gwkokab.analysis.utils.logger) log_info() (in module gwkokab.analysis.utils.logger) log_planck_taper_window() (in module gwkokab.utils) (in module gwkokab.utils.kernel) log_prob() (gwkokab.models.BrokenPowerlaw method) (gwkokab.models.BrokenPowerlawTwoPeak method) (gwkokab.models.GWTC4EffectiveSpinSkewNormalModel method) (gwkokab.models.mass.BrokenPowerlaw method) (gwkokab.models.mass.BrokenPowerlawTwoPeak method) (gwkokab.models.mass.GaussianPrimaryMassRatio method) (gwkokab.models.mass.GenericSmoothedPowerlawMassRatio method) (gwkokab.models.mass.PowerlawPrimaryMassRatio method) (gwkokab.models.mass.SmoothedBrokenPowerlawMassRatioPowerlaw method) (gwkokab.models.mass.SmoothedGaussianPrimaryMassRatio method) (gwkokab.models.mass.SmoothedPowerlawPrimaryMassRatio method) (gwkokab.models.mass.SmoothedTwoComponentPrimaryMassRatio method) (gwkokab.models.mass.Wysocki2019MassModel method) (gwkokab.models.PowerlawPrimaryMassRatio method) (gwkokab.models.SmoothedTwoComponentPrimaryMassRatio method) (gwkokab.models.spin.GWTC4EffectiveSpinSkewNormalModel method) (gwkokab.models.utils.DoublyTruncatedPowerLaw method) (gwkokab.models.utils.JointDistribution method) (gwkokab.models.utils.LazyJointDistribution method) (gwkokab.models.utils.ScaledMixture method) (gwkokab.models.Wysocki2019MassModel method)

log_psi_of_z() (gwkokab.models.MadauDickinsonRedshiftModel method) (gwkokab.models.PowerlawRedshiftModel method) (gwkokab.models.redshift.MadauDickinsonRedshiftModel method) (gwkokab.models.redshift.PowerlawRedshiftModel method) log_scales (gwkokab.models.utils.ScaledMixture attribute) logdVcdz() (gwkokab.cosmology.Cosmology method) LoggedAssertionError LoggedDeprecationWarning LoggedImportError LoggedIndexError LoggedKeyError LoggedMixinException (class in gwkokab.utils.exceptions) LoggedMixinWarning (class in gwkokab.utils.exceptions) LoggedNotImplementedError LoggedRuntimeWarning LoggedTypeError LoggedUserWarning LoggedValueError logging (gwkokab.analysis.core.flowMC_base.Sampler attribute) logsubexp() (in module gwkokab.utils.math) LUMINOSITY_DISTANCE (gwkokab.parameters.Parameters attribute)

M

m1_m2_chi1_chi2_costilt1_costilt2_to_chieff() (in module gwkokab.utils) (in module gwkokab.utils.transformations) m1_m2_chi1_chi2_costilt1_costilt2_to_chiminus() (in module gwkokab.utils) (in module gwkokab.utils.transformations) m1_m2_chi1z_chi2z_to_chiminus() (in module gwkokab.utils) (in module gwkokab.utils.transformations) m1_m2_chieff_chiminus_to_chi1z_chi2z() (in module gwkokab.utils) (in module gwkokab.utils.transformations) m1_q_to_m2() (in module gwkokab.utils) (in module gwkokab.utils.transformations) m1_times_m2() (in module gwkokab.utils) (in module gwkokab.utils.transformations) m2_q_to_m1() (in module gwkokab.utils) (in module gwkokab.utils.transformations) m_det_z_to_m_source() (in module gwkokab.utils) (in module gwkokab.utils.transformations) m_source_z_to_m_det() (in module gwkokab.utils) (in module gwkokab.utils.transformations) MadauDickinsonRedshiftModel (class in gwkokab.models) (class in gwkokab.models.redshift) main() (in module gwkokab.analysis.ecc_matters.analytical) (in module gwkokab.analysis.ecc_matters.discrete) (in module gwkokab.analysis.ecc_matters.synthetic_events) (in module gwkokab.analysis.multisource.analytical) (in module gwkokab.analysis.multisource.discrete) (in module gwkokab.analysis.n_pls_m_gs.analytical) (in module gwkokab.analysis.n_pls_m_gs.discrete) (in module gwkokab.analysis.n_pls_m_gs.synthetic_events) (in module gwkokab.analysis.subpopulation.analytical) (in module gwkokab.analysis.subpopulation.discrete) (in module gwkokab_scripts.batch_scatter2d) (in module gwkokab_scripts.batch_scatter3d) (in module gwkokab_scripts.chain_plot) (in module gwkokab_scripts.corner_overplot) (in module gwkokab_scripts.corner_plot) (in module gwkokab_scripts.diag_mass_matrix) (in module gwkokab_scripts.ess_evolution_plot) (in module gwkokab_scripts.ess_plot) (in module gwkokab_scripts.flowMC_info) (in module gwkokab_scripts.h5repack) (in module gwkokab_scripts.hist_overplot) (in module gwkokab_scripts.joint_plot) (in module gwkokab_scripts.param_lens) (in module gwkokab_scripts.pe_diagnostics) (in module gwkokab_scripts.r_hat_plot) (in module gwkokab_scripts.scatter2d) (in module gwkokab_scripts.scatter3d) (in module gwkokab_scripts.trace_plot) make_model() (in module gwkokab.utils) (in module gwkokab.utils.train) make_parser() (in module gwkokab_scripts.batch_scatter2d) (in module gwkokab_scripts.batch_scatter3d) (in module gwkokab_scripts.chain_plot) (in module gwkokab_scripts.corner_overplot) (in module gwkokab_scripts.corner_plot) (in module gwkokab_scripts.ess_evolution_plot) (in module gwkokab_scripts.ess_plot) (in module gwkokab_scripts.hist_overplot) (in module gwkokab_scripts.joint_plot) (in module gwkokab_scripts.r_hat_plot) (in module gwkokab_scripts.scatter2d) (in module gwkokab_scripts.scatter3d) marginal_distributions (gwkokab.models.utils.JointDistribution attribute) (gwkokab.models.utils.LazyJointDistribution attribute) marginal_log_probs() (gwkokab.models.utils.JointDistribution method) mass_matrix (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) MASS_RATIO (gwkokab.parameters.Parameters attribute) mass_ratio() (in module gwkokab.utils) (in module gwkokab.utils.transformations) mass_ratio_mass_sandwich (in module gwkokab.models.constraints) mass_sandwich (in module gwkokab.models.constraints) match_all() (in module gwkokab.analysis.utils.regex) matches_regex() (in module gwkokab.analysis.utils.regex) max_samples (gwkokab.analysis.core.inference_io.DiscretePELoader attribute) max_tree_depth (gwkokab.analysis.core.inference_io.NumpyroNUTSSamplerConfig attribute) Mc_eta_to_m1_m2() (in module gwkokab.utils) (in module gwkokab.utils.transformations) mcmc (gwkokab.analysis.core.inference_io.NumpyroGlobalConfig attribute) mean (gwkokab.models.utils.ScaledMixture property) MEAN_ANOMALY (gwkokab.parameters.Parameters attribute) mixture_dim (gwkokab.models.utils.ScaledMixture property) mixture_size (gwkokab.models.utils.ScaledMixture property) mock_spin_error() (in module gwkokab.errors) model (gwkokab.analysis.core.analysis_base.AnalysisBase attribute) model_arg_parser() (in module gwkokab.analysis.multisource.common) (in module gwkokab.analysis.n_pls_m_gs.common) (in module gwkokab.analysis.subpopulation.common) model_config (gwkokab.analysis.core.inference_io.AnalyticalPELoader attribute) (gwkokab.analysis.core.inference_io.DiscretePELoader attribute) (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) (gwkokab.analysis.core.inference_io.NumpyroGlobalConfig attribute) (gwkokab.analysis.core.inference_io.NumpyroMCMCConfig attribute) (gwkokab.analysis.core.inference_io.NumpyroNUTSSamplerConfig attribute) model_fn (gwkokab.analysis.core.synthetic_events.SyntheticEventsBase attribute) (gwkokab.analysis.ecc_matters.common.EccentricityMattersCore attribute) (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) model_parameters (gwkokab.analysis.core.analysis_base.AnalysisBase property) (gwkokab.analysis.core.synthetic_events.SyntheticEventsBase property) (gwkokab.analysis.ecc_matters.common.EccentricityMattersCore property) (gwkokab.analysis.multisource.common.MultiSourceModelCore property) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore property) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore property) model_params (gwkokab.analysis.core.synthetic_events.SyntheticEventsBase attribute) ModelRegistry (class in gwkokab.models.wrappers) modify_model_params() (gwkokab.analysis.core.analysis_base.AnalysisBase method) (gwkokab.analysis.core.synthetic_events.SyntheticEventsBase method) (gwkokab.analysis.multisource.common.MultiSourceModelCore method) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore method) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore method)

module generate_report gwkokab gwkokab.analysis gwkokab.analysis.core gwkokab.analysis.core.analysis_base gwkokab.analysis.core.analytical_base gwkokab.analysis.core.discrete_base gwkokab.analysis.core.flowMC_base gwkokab.analysis.core.inference_io gwkokab.analysis.core.numpyro_base gwkokab.analysis.core.synthetic_events gwkokab.analysis.core.synthetic_pe gwkokab.analysis.core.utils gwkokab.analysis.ecc_matters gwkokab.analysis.ecc_matters.analytical gwkokab.analysis.ecc_matters.common gwkokab.analysis.ecc_matters.discrete gwkokab.analysis.ecc_matters.synthetic_events gwkokab.analysis.multisource gwkokab.analysis.multisource.analytical gwkokab.analysis.multisource.common gwkokab.analysis.multisource.discrete gwkokab.analysis.n_pls_m_gs gwkokab.analysis.n_pls_m_gs.analytical gwkokab.analysis.n_pls_m_gs.common gwkokab.analysis.n_pls_m_gs.discrete gwkokab.analysis.n_pls_m_gs.synthetic_events gwkokab.analysis.subpopulation gwkokab.analysis.subpopulation.analytical gwkokab.analysis.subpopulation.common gwkokab.analysis.subpopulation.discrete gwkokab.analysis.utils gwkokab.analysis.utils.checks gwkokab.analysis.utils.common gwkokab.analysis.utils.jenks gwkokab.analysis.utils.literals gwkokab.analysis.utils.logger gwkokab.analysis.utils.marginals gwkokab.analysis.utils.priors gwkokab.analysis.utils.regex gwkokab.constants gwkokab.cosmology gwkokab.errors gwkokab.inference gwkokab.inference.factory gwkokab.inference.flowMC_analytical_poisson_likelihood gwkokab.inference.flowMC_discrete_poisson_likelihood gwkokab.inference.numpyro_analytical_poisson_likelihood gwkokab.inference.numpyro_discrete_poisson_likelihood gwkokab.inference.poissonlikelihood_utils gwkokab.models gwkokab.models.constraints gwkokab.models.hybrids gwkokab.models.mass gwkokab.models.redshift gwkokab.models.spin gwkokab.models.sundry gwkokab.models.transformations gwkokab.models.utils gwkokab.models.wrappers gwkokab.parameters gwkokab.poisson_mean gwkokab.utils gwkokab.utils.exceptions gwkokab.utils.kernel gwkokab.utils.math gwkokab.utils.path gwkokab.utils.tools gwkokab.utils.train gwkokab.utils.transformations gwkokab.version gwkokab_scripts gwkokab_scripts.batch_scatter2d gwkokab_scripts.batch_scatter3d gwkokab_scripts.chain_plot gwkokab_scripts.corner_overplot gwkokab_scripts.corner_plot gwkokab_scripts.diag_mass_matrix gwkokab_scripts.ess_evolution_plot gwkokab_scripts.ess_plot gwkokab_scripts.flowMC_info gwkokab_scripts.h5repack gwkokab_scripts.hist_overplot gwkokab_scripts.joint_plot gwkokab_scripts.param_lens gwkokab_scripts.pe_diagnostics gwkokab_scripts.r_hat_plot gwkokab_scripts.scatter2d gwkokab_scripts.scatter3d gwkokab_scripts.trace_plot mse_loss_fn() (in module gwkokab.utils) (in module gwkokab.utils.train) MSUN_SI (in module gwkokab.constants) MultiSourceModel() (in module gwkokab.models) (in module gwkokab.models.hybrids) MultiSourceModelAnalyticalAnalysis (class in gwkokab.analysis.multisource.analytical) MultiSourceModelCore (class in gwkokab.analysis.multisource.common) MultiSourceModelDiscreteAnalysis (class in gwkokab.analysis.multisource.discrete) MultiSourceModelFAnalyticalAnalysis (class in gwkokab.analysis.multisource.analytical) MultiSourceModelFDiscreteAnalysis (class in gwkokab.analysis.multisource.discrete) MultiSourceModelNAnalyticalAnalysis (class in gwkokab.analysis.multisource.analytical) MultiSourceModelNDiscreteAnalysis (class in gwkokab.analysis.multisource.discrete)

N

N_bpl (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) n_buckets (gwkokab.analysis.core.discrete_base.DiscreteBase attribute) n_buffer_events (gwkokab.analysis.core.synthetic_events.SyntheticEventsBase attribute) n_chains (gwkokab.analysis.core.flowMC_base.Sampler attribute) (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) n_dim (gwkokab.analysis.core.flowMC_base.Sampler attribute) n_epochs (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) N_g (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) N_gg (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) n_global_steps (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) N_gpl (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) n_leapfrog (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) n_local_steps (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) n_max_examples (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) n_NFproposal_batch_size (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) N_pl (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) n_production_loops (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) n_samples (gwkokab.analysis.core.analytical_base.AnalyticalBase attribute) N_spl (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) n_training_loops (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) NDIsotropicAndTruncatedNormalMixture() (in module gwkokab.models) (in module gwkokab.models.sundry)

NDTwoTruncatedNormalMixture() (in module gwkokab.models.sundry) normalize_path() (in module gwkokab.utils.path) NPowerlawMGaussian() (in module gwkokab.models) (in module gwkokab.models.hybrids) NPowerlawMGaussianAnalyticalAnalysis (class in gwkokab.analysis.n_pls_m_gs.analytical) NPowerlawMGaussianCore (class in gwkokab.analysis.n_pls_m_gs.common) NPowerlawMGaussianDiscreteAnalysis (class in gwkokab.analysis.n_pls_m_gs.discrete) NPowerlawMGaussianFAnalyticalAnalysis (class in gwkokab.analysis.n_pls_m_gs.analytical) NPowerlawMGaussianFDiscreteAnalysis (class in gwkokab.analysis.n_pls_m_gs.discrete) NPowerlawMGaussianNAnalyticalAnalysis (class in gwkokab.analysis.n_pls_m_gs.analytical) NPowerlawMGaussianNDiscreteAnalysis (class in gwkokab.analysis.n_pls_m_gs.discrete) num_chains (gwkokab.analysis.core.inference_io.NumpyroMCMCConfig attribute) num_samples (gwkokab.analysis.core.inference_io.NumpyroMCMCConfig attribute) num_warmup (gwkokab.analysis.core.inference_io.NumpyroMCMCConfig attribute) numpyro_analytical_poisson_likelihood() (in module gwkokab.inference) (in module gwkokab.inference.numpyro_analytical_poisson_likelihood) numpyro_arg_parser (in module gwkokab.analysis.core.numpyro_base) numpyro_discrete_poisson_likelihood() (in module gwkokab.inference) (in module gwkokab.inference.numpyro_discrete_poisson_likelihood) NumpyroBase (class in gwkokab.analysis.core.numpyro_base) NumpyroGlobalConfig (class in gwkokab.analysis.core.inference_io) NumpyroMCMCConfig (class in gwkokab.analysis.core.inference_io) NumpyroNUTSSamplerConfig (class in gwkokab.analysis.core.inference_io)

O

OmegaKappa (gwkokab.cosmology.Cosmology attribute) OmegaLambda (gwkokab.cosmology.Cosmology attribute)

OmegaMatter (gwkokab.cosmology.Cosmology attribute) OmegaRadiation (gwkokab.cosmology.Cosmology attribute) outdir (gwkokab.analysis.core.flowMC_base.Sampler attribute)

P

pad_and_stack() (in module gwkokab.analysis.utils.jenks) parameter_aliases (gwkokab.analysis.core.inference_io.AnalyticalPELoader attribute) (gwkokab.analysis.core.inference_io.DiscretePELoader attribute) Parameters (class in gwkokab.parameters) parameters (gwkokab.analysis.core.analysis_base.AnalysisBase property) (gwkokab.analysis.core.synthetic_events.SyntheticEventsBase property) (gwkokab.analysis.ecc_matters.common.EccentricityMattersCore property) (gwkokab.analysis.multisource.common.MultiSourceModelCore property) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore property) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore property) partial_order (gwkokab.models.utils.LazyJointDistribution attribute) PHI_1 (gwkokab.parameters.Parameters attribute) PHI_12 (gwkokab.parameters.Parameters attribute) PHI_2 (gwkokab.parameters.Parameters attribute) PHI_ORB (gwkokab.parameters.Parameters attribute) PLANCK_2013_Cosmology() (in module gwkokab.cosmology) PLANCK_2015_Cosmology() (in module gwkokab.cosmology) PLANCK_2018_Cosmology() (in module gwkokab.cosmology) plot_history() (in module gwkokab_scripts.flowMC_info) plot_marginal_with_intervals() (in module gwkokab.analysis.utils.marginals) PlotStyle (class in gwkokab.analysis.utils.marginals) poisson_mean_filename (gwkokab.analysis.core.analysis_base.AnalysisBase attribute) (gwkokab.analysis.core.synthetic_events.SyntheticEventsBase attribute) poisson_mean_from_neural_pdet() (in module gwkokab.poisson_mean) poisson_mean_from_neural_vt() (in module gwkokab.poisson_mean) poisson_mean_from_sensitivity_injections() (in module gwkokab.poisson_mean) PoissonMeanEstimationLoader (class in gwkokab.analysis.core.inference_io)

polar_to_cart() (in module gwkokab.utils) (in module gwkokab.utils.transformations) POLARIZATION_ANGLE (gwkokab.parameters.Parameters attribute) positive_decreasing_vector (in module gwkokab.models.constraints) positive_increasing_vector (in module gwkokab.models.constraints) PowerlawPrimaryMassRatio (class in gwkokab.models) (class in gwkokab.models.mass) PowerlawRedshiftModel (class in gwkokab.models) (class in gwkokab.models.redshift) PRECESSING_SPIN (gwkokab.parameters.Parameters attribute) predict() (in module gwkokab.utils) (in module gwkokab.utils.train) PRIMARY_MASS_DETECTED (gwkokab.parameters.Parameters attribute) primary_mass_distribution (gwkokab.models.mass.GenericSmoothedPowerlawMassRatio attribute) PRIMARY_MASS_SOURCE (gwkokab.parameters.Parameters attribute) PRIMARY_SPIN_MAGNITUDE (gwkokab.parameters.Parameters attribute) PRIMARY_SPIN_X (gwkokab.parameters.Parameters attribute) PRIMARY_SPIN_Y (gwkokab.parameters.Parameters attribute) PRIMARY_SPIN_Z (gwkokab.parameters.Parameters attribute) PrimaryMassAndMassRatioToComponentMassesTransform (class in gwkokab.models.transformations) print_header() (in module gwkokab_scripts.flowMC_info) print_section() (in module gwkokab_scripts.flowMC_info) prior_filename (gwkokab.analysis.core.analysis_base.AnalysisBase attribute) PRNGKeyMixin (class in gwkokab.analysis.core.utils) profile_memory (gwkokab.analysis.core.analysis_base.AnalysisBase attribute) progress_bar (gwkokab.analysis.core.inference_io.NumpyroMCMCConfig attribute) progress_rate (gwkokab.analysis.core.inference_io.NumpyroMCMCConfig attribute)

Q

q_from_eta() (in module gwkokab.utils.transformations)

R

read_attrs_from_hdf5() (in module gwkokab.analysis.core.utils) read_data() (gwkokab.analysis.core.discrete_base.DiscreteBase method) (in module gwkokab.utils) (in module gwkokab.utils.train) read_domains() (in module gwkokab.analysis.utils.marginals) read_from_hdf5() (in module gwkokab.analysis.core.utils) read_json() (in module gwkokab.analysis.utils.common) REDSHIFT (gwkokab.parameters.Parameters attribute) RedshiftToLuminosityDistance (class in gwkokab.models.transformations) REDUCED_MASS (gwkokab.parameters.Parameters attribute) reduced_mass() (in module gwkokab.utils) (in module gwkokab.utils.transformations) register() (gwkokab.models.wrappers.ModelRegistry method) registry (gwkokab.models.wrappers.ModelRegistry property) regularize_mass_matrix (gwkokab.analysis.core.inference_io.NumpyroNUTSSamplerConfig attribute) RelationMesh (class in gwkokab.parameters)

remove_comoving_volume_factor() (in module gwkokab.analysis.utils.marginals) resolve() (gwkokab.parameters.RelationMesh method) resolve_from_arrays() (gwkokab.parameters.RelationMesh method) resources (gwkokab.analysis.core.flowMC_base.Local_Global_Sampler_Bundle attribute) (gwkokab.analysis.core.flowMC_base.Sampler attribute) RIGHT_ASCENSION (gwkokab.parameters.Parameters attribute) rng_key (gwkokab.analysis.core.flowMC_base.Sampler attribute) (gwkokab.analysis.core.utils.PRNGKeyMixin property) root_dir (gwkokab.analysis.core.synthetic_pe.SyntheticDiscretePE attribute) rq_spline_hidden_units (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) rq_spline_n_bins (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) rq_spline_n_layers (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) rq_spline_range (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) rules (gwkokab.parameters.RelationMesh attribute) run() (gwkokab.analysis.core.analytical_base.AnalyticalBase method) (gwkokab.analysis.core.discrete_base.DiscreteBase method)

S

safe_filename() (in module gwkokab_scripts.pe_diagnostics) sample() (gwkokab.analysis.core.flowMC_base.Sampler method) (gwkokab.models.mass.PowerlawPrimaryMassRatio method) (gwkokab.models.mass.Wysocki2019MassModel method) (gwkokab.models.PowerlawPrimaryMassRatio method) (gwkokab.models.utils.DoublyTruncatedPowerLaw method) (gwkokab.models.utils.JointDistribution method) (gwkokab.models.utils.LazyJointDistribution method) (gwkokab.models.utils.ScaledMixture method) (gwkokab.models.Wysocki2019MassModel method) sample_transformer (gwkokab.analysis.core.inference_io.AnalyticalPELoader attribute) sample_with_intermediates() (gwkokab.models.utils.ScaledMixture method) Sampler (class in gwkokab.analysis.core.flowMC_base) sampler_cfg (gwkokab.analysis.core.analysis_base.AnalysisBase attribute) sampler_name (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) (gwkokab.analysis.core.inference_io.NumpyroGlobalConfig attribute) SamplerConfig (class in gwkokab.analysis.core.inference_io) SAMPLES_GROUP_NAME (in module gwkokab.analysis.utils.literals) SampleTransformer (class in gwkokab.analysis.core.utils) save_autocorr_plot() (in module gwkokab_scripts.pe_diagnostics) save_hist_plot() (in module gwkokab_scripts.pe_diagnostics) save_model() (in module gwkokab.utils) (in module gwkokab.utils.train) save_population() (gwkokab.analysis.core.synthetic_events.SyntheticEventsBase method) save_results_to_hdf5() (in module gwkokab.analysis.utils.marginals) save_trace_plot() (in module gwkokab_scripts.pe_diagnostics) ScaledMixture (class in gwkokab.models.utils) SECONDARY_MASS_DETECTED (gwkokab.parameters.Parameters attribute) SECONDARY_MASS_SOURCE (gwkokab.parameters.Parameters attribute) SECONDARY_SPIN_MAGNITUDE (gwkokab.parameters.Parameters attribute) SECONDARY_SPIN_X (gwkokab.parameters.Parameters attribute) SECONDARY_SPIN_Y (gwkokab.parameters.Parameters attribute) SECONDARY_SPIN_Z (gwkokab.parameters.Parameters attribute) SECONDS_PER_YEAR (in module gwkokab.constants) seed (gwkokab.analysis.core.utils.PRNGKeyMixin property) serialize() (gwkokab.analysis.core.flowMC_base.Sampler method) set_log_level() (in module gwkokab.analysis.utils.logger) shaped_values (gwkokab.models.utils.JointDistribution attribute) (gwkokab.models.utils.LazyJointDistribution attribute) SIN_DECLINATION (gwkokab.parameters.Parameters attribute) sin_tilt() (in module gwkokab.utils) (in module gwkokab.utils.transformations) size (gwkokab.analysis.core.synthetic_pe.SyntheticDiscretePE attribute) SmoothedBrokenPowerlawMassRatioPowerlaw (class in gwkokab.models.mass) SmoothedGaussianPrimaryMassRatio (class in gwkokab.models.mass) SmoothedPowerlawPrimaryMassRatio (class in gwkokab.models.mass) SmoothedTwoComponentPrimaryMassRatio (class in gwkokab.models) (class in gwkokab.models.mass) SourceMassAndRedshiftToDetectedMassAndRedshift (class in gwkokab.models.transformations) SPEED_OF_LIGHT (in module gwkokab.constants) spherical_to_cart() (in module gwkokab.utils) (in module gwkokab.utils.transformations) spin_costilt_from_components() (in module gwkokab.utils) (in module gwkokab.utils.transformations)

spin_magnitude_from_components() (in module gwkokab.utils) (in module gwkokab.utils.transformations) split_chains() (in module gwkokab_scripts.pe_diagnostics) split_single_chain_for_rhat() (in module gwkokab_scripts.pe_diagnostics) step_size (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute) (gwkokab.analysis.core.inference_io.NumpyroNUTSSamplerConfig attribute) strategies (gwkokab.analysis.core.flowMC_base.Local_Global_Sampler_Bundle attribute) (gwkokab.analysis.core.flowMC_base.Sampler attribute) strategy_order (gwkokab.analysis.core.flowMC_base.Local_Global_Sampler_Bundle attribute) (gwkokab.analysis.core.flowMC_base.Sampler attribute) strictly_decreasing_vector (in module gwkokab.models.constraints) strictly_increasing_vector (in module gwkokab.models.constraints) SubPopulationModel() (in module gwkokab.models) (in module gwkokab.models.hybrids) SubPopulationModelAnalyticalAnalysis (class in gwkokab.analysis.subpopulation.analytical) SubPopulationModelCore (class in gwkokab.analysis.subpopulation.common) SubPopulationModelDiscreteAnalysis (class in gwkokab.analysis.subpopulation.discrete) SubPopulationModelFAnalyticalAnalysis (class in gwkokab.analysis.subpopulation.analytical) SubPopulationModelFDiscreteAnalysis (class in gwkokab.analysis.subpopulation.discrete) SubPopulationModelNAnalyticalAnalysis (class in gwkokab.analysis.subpopulation.analytical) SubPopulationModelNDiscreteAnalysis (class in gwkokab.analysis.subpopulation.discrete) support (gwkokab.models.GWTC4EffectiveSpinSkewNormalModel attribute) (gwkokab.models.spin.GWTC4EffectiveSpinSkewNormalModel attribute) (gwkokab.models.utils.ExtendedSupportTransformedDistribution property) support() (gwkokab.models.BrokenPowerlaw method) (gwkokab.models.BrokenPowerlawTwoPeak method) (gwkokab.models.mass.BrokenPowerlaw method) (gwkokab.models.mass.BrokenPowerlawTwoPeak method) (gwkokab.models.mass.GaussianPrimaryMassRatio method) (gwkokab.models.mass.GenericSmoothedPowerlawMassRatio method) (gwkokab.models.mass.PowerlawPrimaryMassRatio method) (gwkokab.models.mass.SmoothedBrokenPowerlawMassRatioPowerlaw method) (gwkokab.models.mass.SmoothedGaussianPrimaryMassRatio method) (gwkokab.models.mass.SmoothedPowerlawPrimaryMassRatio method) (gwkokab.models.mass.SmoothedTwoComponentPrimaryMassRatio method) (gwkokab.models.mass.Wysocki2019MassModel method) (gwkokab.models.PowerlawPrimaryMassRatio method) (gwkokab.models.SmoothedTwoComponentPrimaryMassRatio method) (gwkokab.models.utils.DoublyTruncatedPowerLaw method) (gwkokab.models.utils.JointDistribution method) (gwkokab.models.utils.LazyJointDistribution method) (gwkokab.models.utils.ScaledMixture method) (gwkokab.models.Wysocki2019MassModel method) SYMMETRIC_MASS_RATIO (gwkokab.parameters.Parameters attribute) symmetric_mass_ratio() (in module gwkokab.utils) (in module gwkokab.utils.transformations) symmetric_mass_ratio_to_delta_m() (in module gwkokab.utils) (in module gwkokab.utils.transformations) synthetic_analytical_pe_main() (in module gwkokab.analysis.core.synthetic_pe) synthetic_discrete_pe_main() (in module gwkokab.analysis.core.synthetic_pe) SyntheticAnalyticalPE (class in gwkokab.analysis.core.synthetic_pe) SyntheticDiscretePE (class in gwkokab.analysis.core.synthetic_pe) SyntheticEventsBase (class in gwkokab.analysis.core.synthetic_events)

T

target_accept_prob (gwkokab.analysis.core.inference_io.NumpyroNUTSSamplerConfig attribute) thinning (gwkokab.analysis.core.inference_io.NumpyroMCMCConfig attribute) threshold (gwkokab.analysis.core.discrete_base.DiscreteBase attribute) time_now() (in module gwkokab.analysis.utils.logger) to_structured() (in module gwkokab.analysis.core.utils) TOTAL_MASS (gwkokab.parameters.Parameters attribute) total_mass() (in module gwkokab.utils) (in module gwkokab.utils.transformations) trace_plot() (in module gwkokab_scripts.trace_plot)

train_regressor() (in module gwkokab.utils) (in module gwkokab.utils.train) transform() (gwkokab.analysis.core.utils.IdentitySampleTransformer method) (gwkokab.analysis.core.utils.SampleTransformer method) transforms (gwkokab.models.transformations.BlockTransform attribute) truncated_normal_error() (in module gwkokab.errors) truncnorm_logpdf() (in module gwkokab.utils.math) TwoTruncatedNormalMixture() (in module gwkokab.models) (in module gwkokab.models.sundry)

U

use_beta_spin_magnitude (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) use_chi_eff_mixture (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) use_cos_iota (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) use_detection_time (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) use_eccentricity_mixture (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) use_eccentricity_powerlaw (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) use_madau_dickinson_redshift (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) use_mean_anomaly (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) use_phi_1 (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) use_phi_12 (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) use_phi_2 (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) use_phi_orb (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) use_polarization_angle (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute)

use_powerlaw_redshift (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) use_right_ascension (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) use_sin_declination (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) use_skew_normal_chi_eff (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) use_spin_magnitude_mixture (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) use_tilt (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) use_truncated_normal_chi_p (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) use_truncated_normal_spin_x (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) use_truncated_normal_spin_y (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute) use_truncated_normal_spin_z (gwkokab.analysis.multisource.common.MultiSourceModelCore attribute) (gwkokab.analysis.n_pls_m_gs.common.NPowerlawMGaussianCore attribute) (gwkokab.analysis.subpopulation.common.SubPopulationModelCore attribute)

V

variance (gwkokab.models.utils.ScaledMixture property) variance_cut_threshold (gwkokab.analysis.core.analysis_base.AnalysisBase attribute)

Vc (gwkokab.cosmology.Cosmology attribute) verbose (gwkokab.analysis.core.flowMC_base.Sampler attribute) (gwkokab.analysis.core.inference_io.FlowMCGlobalConfig attribute)

W

waveform_name (gwkokab.analysis.core.synthetic_pe.SyntheticAnalyticalPE attribute) (gwkokab.analysis.core.synthetic_pe.SyntheticDiscretePE attribute) where_fns (gwkokab.analysis.core.discrete_base.DiscreteBase attribute) where_fns_list() (in module gwkokab.analysis.multisource.common) (in module gwkokab.analysis.n_pls_m_gs.common) (in module gwkokab.analysis.subpopulation.common)

write_domains() (in module gwkokab.analysis.utils.marginals) write_json() (in module gwkokab.analysis.utils.common) write_summary() (in module gwkokab_scripts.pe_diagnostics) write_to_hdf5() (in module gwkokab.analysis.core.utils) Wysocki2019MassModel (class in gwkokab.models) (class in gwkokab.models.mass)

Z

z (gwkokab.cosmology.Cosmology property) z_to_Dc() (gwkokab.cosmology.Cosmology method)

z_to_DL() (gwkokab.cosmology.Cosmology method) z_to_E() (gwkokab.cosmology.Cosmology method)