Index

A

adaptive_histogram_skymap() (in module amplfi.utils.skymap) add_arguments_to_parser() (amplfi.train.cli.base.AmplfiBaseCLI method) (amplfi.train.cli.flow.AmplfiFlowCLI method) (amplfi.train.cli.similarity.AmplfiSimilarityCLI method) after_instantiate_classes() (amplfi.train.cli.base.AmplfiBaseCLI method) amplfi module amplfi.train.architectures.embeddings module amplfi.train.architectures.flows module amplfi.train.augmentations module amplfi.train.callbacks module amplfi.train.cli.base module amplfi.train.cli.flow module amplfi.train.cli.similarity module

amplfi.train.losses module amplfi.train.models.flow module amplfi.train.models.similarity module amplfi.train.prior module amplfi.utils.distance module amplfi.utils.result module amplfi.utils.skymap module AmplfiBaseCLI (class in amplfi.train.cli.base) AmplfiFlowCLI (class in amplfi.train.cli.flow) AmplfiPrior (class in amplfi.train.prior) AmplfiResult (class in amplfi.utils.result) AmplfiSimilarityCLI (class in amplfi.train.cli.similarity) analyze_event() (amplfi.train.models.flow.FlowModel method) ansatz_impl() (in module amplfi.utils.distance)

B

build_flow() (amplfi.train.architectures.flows.FlowArchitecture method) (amplfi.train.architectures.flows.NSF method)

C

cast_as_bilby_result() (amplfi.train.models.flow.FlowModel method) configure_callbacks() (amplfi.train.models.flow.FlowModel method) (amplfi.train.models.similarity.SimilarityModel method)

covariance_loss() (in module amplfi.train.losses)

D

dHdz() (in module amplfi.utils.distance)

E

eps (amplfi.train.losses.VICRegLoss attribute)

F

f() (in module amplfi.utils.distance) filter_parameters() (amplfi.train.models.flow.FlowModel method) FlowArchitecture (class in amplfi.train.architectures.flows) FlowModel (class in amplfi.train.models.flow) forward() (amplfi.train.augmentations.PsdEstimator method) (amplfi.train.augmentations.WaveformProjector method) (amplfi.train.losses.VICRegLoss method) (amplfi.train.models.flow.FlowModel method) (amplfi.train.models.similarity.SimilarityModel method) (amplfi.train.prior.ParameterTransformer method)

fprime() (in module amplfi.utils.distance)

H

H() (in module amplfi.utils.distance)

histogram_skymap() (in module amplfi.utils.skymap)

I

invariance_loss() (in module amplfi.train.losses)

L

lambda_param (amplfi.train.losses.VICRegLoss attribute)

log_prob() (amplfi.train.architectures.flows.FlowArchitecture method) (amplfi.train.prior.AmplfiPrior method)

M

main() (in module amplfi.train.cli.flow) (in module amplfi.train.cli.similarity) module amplfi amplfi.train.architectures.embeddings amplfi.train.architectures.flows amplfi.train.augmentations amplfi.train.callbacks amplfi.train.cli.base amplfi.train.cli.flow amplfi.train.cli.similarity amplfi.train.losses amplfi.train.models.flow amplfi.train.models.similarity amplfi.train.prior amplfi.utils.distance amplfi.utils.result amplfi.utils.skymap

moments_from_samples_impl() (in module amplfi.utils.distance) mu_param (amplfi.train.losses.VICRegLoss attribute)

N

nest2uniq() (in module amplfi.utils.skymap)

NSF (class in amplfi.train.architectures.flows) nu_param (amplfi.train.losses.VICRegLoss attribute)

O

on_predict_epoch_start() (amplfi.train.models.flow.FlowModel method)

on_test_batch_end() (amplfi.train.models.flow.FlowModel method) on_test_epoch_start() (amplfi.train.models.flow.FlowModel method)

P

P() (in module amplfi.utils.distance) ParameterTransformer (class in amplfi.train.prior)

plot_skymap() (in module amplfi.utils.skymap) predict_step() (amplfi.train.models.flow.FlowModel method) PsdEstimator (class in amplfi.train.augmentations)

Q

Q() (in module amplfi.utils.distance)

R

reweight_to_prior() (amplfi.utils.result.AmplfiResult method)

S

sample() (amplfi.train.architectures.flows.FlowArchitecture method)

SimilarityModel (class in amplfi.train.models.similarity)

T

test_step() (amplfi.train.models.flow.FlowModel method) to_crossmatch_result() (amplfi.utils.result.AmplfiResult method)

to_skymap() (amplfi.utils.result.AmplfiResult method) training_step() (amplfi.train.models.flow.FlowModel method) (amplfi.train.models.similarity.SimilarityModel method)

V

validation_step() (amplfi.train.models.flow.FlowModel method) (amplfi.train.models.similarity.SimilarityModel method)

variance_loss() (in module amplfi.train.losses) VICRegLoss (class in amplfi.train.losses)

W

WaveformProjector (class in amplfi.train.augmentations)

X

x2() (in module amplfi.utils.distance) x2_prime() (in module amplfi.utils.distance) x3() (in module amplfi.utils.distance)

x3_prime() (in module amplfi.utils.distance) x4() (in module amplfi.utils.distance) x4_prime() (in module amplfi.utils.distance)