Index

_

__call__() (torch_mist.decomposition.base.CenterAndScale method) (torch_mist.distributions.parametrizations.map.ParameterMap method) __del__() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) (torch_mist.utils.train.utils.RunTerminationManager method) __getitem__() (torch_mist.utils.data.csv.CSVDataset method) (torch_mist.utils.data.CSVDataset method) (torch_mist.utils.data.dataset.DataFrameDataset method) (torch_mist.utils.data.dataset.DistributionDataset method) (torch_mist.utils.data.dataset.SampleDataset method) (torch_mist.utils.data.dataset.WrappedDataset method) (torch_mist.utils.data.SampleDataset method) __iter__() (torch_mist.utils.data.SameAttributeSampler method) (torch_mist.utils.data.sampler.SameAttributeSampler method) __len__() (torch_mist.utils.data.csv.CSVDataset method) (torch_mist.utils.data.CSVDataset method) (torch_mist.utils.data.dataset.DataFrameDataset method) (torch_mist.utils.data.dataset.DistributionDataset method) (torch_mist.utils.data.dataset.SampleDataset method) (torch_mist.utils.data.dataset.WrappedDataset method) (torch_mist.utils.data.SameAttributeSampler method) (torch_mist.utils.data.SampleDataset method) (torch_mist.utils.data.sampler.SameAttributeSampler method) __repr__() (torch_mist.distributions.caching.CachedConditionalTransformModule method) (torch_mist.distributions.caching.CachedTransformModule method) (torch_mist.distributions.categorical.CategoricalModule method) (torch_mist.distributions.CategoricalModule method) (torch_mist.distributions.conditional.unconditional.UnconditionalDistribution method) (torch_mist.distributions.empirical.EmpiricalDistribution method) (torch_mist.distributions.EmpiricalDistribution method) (torch_mist.distributions.joint.base.ConditionalNamedDistribution method) (torch_mist.distributions.joint.base.ConditionedJointDistribution method) (torch_mist.distributions.normal.NormalModule method) (torch_mist.distributions.NormalModule method) (torch_mist.distributions.transforms.base.ConditionalTransformedDistributionModule method) (torch_mist.distributions.transforms.base.DistributionModule method) (torch_mist.distributions.transforms.base.TransformedDistributionModule method) (torch_mist.distributions.transforms.ConditionalTransformedDistributionModule method) (torch_mist.distributions.transforms.DistributionModule method) (torch_mist.distributions.transforms.TransformedDistributionModule method) (torch_mist.estimators.discriminative.base.BaselineDiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.base.DiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.BaselineDiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.DiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.FLO method), [1] (torch_mist.estimators.discriminative.implementations.FLO method) (torch_mist.estimators.discriminative.implementations.flo.FLO method) (torch_mist.estimators.DiscriminativeMIEstimator method) (torch_mist.estimators.DoE method), [1] (torch_mist.estimators.FLO method), [1] (torch_mist.estimators.generative.base.ConditionalGenerativeMIEstimator method) (torch_mist.estimators.generative.base.JointGenerativeMIEstimator method) (torch_mist.estimators.generative.ConditionalGenerativeMIEstimator method) (torch_mist.estimators.generative.DoE method), [1] (torch_mist.estimators.generative.implementations.DoE method) (torch_mist.estimators.generative.implementations.doe.DoE method) (torch_mist.estimators.hybrid.base.HybridMIEstimator method) (torch_mist.estimators.hybrid.DiscriminativeMIEstimator method), [1] (torch_mist.estimators.hybrid.HybridMIEstimator method) (torch_mist.estimators.HybridMIEstimator method) (torch_mist.estimators.transformed.utils.DummyModule method) (torch_mist.nn.CachedModule method) __version__ (in module torch_mist) _add_default_conditional_dist_params() (torch_mist.decomposition.CEB method) (torch_mist.decomposition.ceb.CEB method) (torch_mist.decomposition.TIB method) (torch_mist.decomposition.tib.TIB method) _add_default_mi_estimator_params() (torch_mist.decomposition.CEB method) (torch_mist.decomposition.ceb.CEB method) (torch_mist.decomposition.MIB method) (torch_mist.decomposition.mib.MIB method) (torch_mist.decomposition.VIB method) (torch_mist.decomposition.vib.VIB method) _add_default_model_params() (torch_mist.decomposition.base.DimensionalityReduction method) (torch_mist.decomposition.CEB method) (torch_mist.decomposition.ceb.CEB method) (torch_mist.decomposition.MIB method) (torch_mist.decomposition.mib.MIB method) (torch_mist.decomposition.MID method) (torch_mist.decomposition.mid.MID method) (torch_mist.decomposition.TIB method) (torch_mist.decomposition.tib.TIB method) (torch_mist.decomposition.VIB method) (torch_mist.decomposition.vib.VIB method) _add_default_proj_params() (torch_mist.decomposition.base.DimensionalityReduction method) (torch_mist.decomposition.CEB method) (torch_mist.decomposition.ceb.CEB method) (torch_mist.decomposition.MIB method) (torch_mist.decomposition.mib.MIB method) (torch_mist.decomposition.MID method) (torch_mist.decomposition.mid.MID method) (torch_mist.decomposition.TIB method) (torch_mist.decomposition.tib.TIB method) (torch_mist.decomposition.VIB method) (torch_mist.decomposition.vib.VIB method) _add_default_train_params() (torch_mist.decomposition.base.DimensionalityReduction method) _add_default_y_proj_params() (torch_mist.decomposition.base.DimensionalityReduction method) (torch_mist.decomposition.MIB method) (torch_mist.decomposition.mib.MIB method) _approx_log_partition() (torch_mist.estimators.discriminative.base.BaselineDiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.base.DiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.BaselineDiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.DiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.DummyDiscriminativeMIEstimator method), [1] (torch_mist.estimators.discriminative.FLO method), [1] (torch_mist.estimators.discriminative.implementations.dummy.DummyDiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.implementations.DummyDiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.implementations.FLO method) (torch_mist.estimators.discriminative.implementations.flo.FLO method) (torch_mist.estimators.discriminative.implementations.InfoNCE method) (torch_mist.estimators.discriminative.implementations.infonce.InfoNCE method) (torch_mist.estimators.discriminative.implementations.SMILE method) (torch_mist.estimators.discriminative.implementations.smile.SMILE method) (torch_mist.estimators.discriminative.InfoNCE method), [1] (torch_mist.estimators.discriminative.SMILE method), [1] (torch_mist.estimators.DiscriminativeMIEstimator method) (torch_mist.estimators.DummyDiscriminativeMIEstimator method), [1] (torch_mist.estimators.FLO method), [1] (torch_mist.estimators.hybrid.DiscriminativeMIEstimator method), [1] (torch_mist.estimators.InfoNCE method), [1] (torch_mist.estimators.SMILE method), [1] _approximate_estimates() (torch_mist.data.multimixture.MultivariateCorrelatedNormalMixture method) _broadcast_log_p_y_given_x() (torch_mist.estimators.generative.implementations.L1Out method) (torch_mist.estimators.generative.implementations.l1out.L1Out method) (torch_mist.estimators.generative.L1Out method), [1] (torch_mist.estimators.L1Out method), [1] _call() (torch_mist.distributions.caching.CachedTransformModule method) (torch_mist.distributions.transforms.EMANormalize method), [1] (torch_mist.distributions.transforms.implementations.EMANormalize method) (torch_mist.distributions.transforms.implementations.linear.ConditionedLinear method) (torch_mist.distributions.transforms.implementations.normalize.EMANormalize method) (torch_mist.distributions.transforms.implementations.Permute method) (torch_mist.distributions.transforms.implementations.permute.Permute method) (torch_mist.distributions.transforms.Permute method), [1]

_determine_train_duration() (in module torch_mist.utils.estimation) _dict_to_tensor() (torch_mist.distributions.joint.categorical.JointCategorical method) (torch_mist.distributions.joint.wrapper.TorchJointDistribution method) _encode() (torch_mist.decomposition.base.DimensionalityReduction method) (torch_mist.decomposition.base.StochasticDimensionalityReduction method) _entropy() (torch_mist.data.multimixture.MultivariateCorrelatedNormalMixture method) (torch_mist.data.multivariate.JointMultivariateNormal method) (torch_mist.distributions.joint.base.JointDistribution method) (torch_mist.distributions.joint.categorical.JointCategorical method) (torch_mist.distributions.joint.wrapper.TorchJointDistribution method) (torch_mist.distributions.JointDistribution method) (torch_mist.estimators.generative.JointDistribution method) (torch_mist.estimators.JointDistribution method) _fit() (torch_mist.quantization.ClusterQuantization method) (torch_mist.quantization.functions.ClusterQuantization method) (torch_mist.quantization.functions.LearnableQuantization method) (torch_mist.quantization.LearnableQuantization method) _get_transformed_y_dim() (torch_mist.decomposition.base.DimensionalityReduction method) (torch_mist.decomposition.base.StochasticDimensionalityReduction method) _instantiate_estimator() (in module torch_mist.utils.estimation) _instantiate_model() (torch_mist.decomposition.base.DimensionalityReduction method) (torch_mist.decomposition.CEB method) (torch_mist.decomposition.ceb.CEB method) (torch_mist.decomposition.MIB method) (torch_mist.decomposition.mib.MIB method) (torch_mist.decomposition.MID method) (torch_mist.decomposition.mid.MID method) (torch_mist.decomposition.TIB method) (torch_mist.decomposition.tib.TIB method) (torch_mist.decomposition.tinfomax.TInfoMax method) (torch_mist.decomposition.VIB method) (torch_mist.decomposition.vib.VIB method) _instantiate_proj() (torch_mist.decomposition.base.DimensionalityReduction method) (torch_mist.decomposition.base.StochasticDimensionalityReduction method) (torch_mist.decomposition.MID method) (torch_mist.decomposition.mid.MID method) _instantiate_y_proj() (torch_mist.decomposition.base.DimensionalityReduction method) (torch_mist.decomposition.CEB method) (torch_mist.decomposition.ceb.CEB method) (torch_mist.decomposition.MIB method) (torch_mist.decomposition.mib.MIB method) (torch_mist.decomposition.MID method) (torch_mist.decomposition.mid.MID method) (torch_mist.decomposition.VIB method) (torch_mist.decomposition.vib.VIB method) _inverse() (torch_mist.distributions.caching.CachedTransformModule method) (torch_mist.distributions.transforms.EMANormalize method), [1] (torch_mist.distributions.transforms.implementations.EMANormalize method) (torch_mist.distributions.transforms.implementations.linear.ConditionedLinear method) (torch_mist.distributions.transforms.implementations.normalize.EMANormalize method) (torch_mist.distributions.transforms.implementations.Permute method) (torch_mist.distributions.transforms.implementations.permute.Permute method) (torch_mist.distributions.transforms.Permute method), [1] _is_cache_valid() (in module torch_mist.utils.caching) _log() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.DummyLogger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) (torch_mist.utils.logging.logger.pandas.PandasLogger method) (torch_mist.utils.logging.logger.PandasLogger method) (torch_mist.utils.logging.logger.wandb.WandbLogger method) (torch_mist.utils.logging.PandasLogger method) _log_buffer() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) _log_prob() (torch_mist.distributions.joint.base.ConditionedJointDistribution method) (torch_mist.distributions.joint.base.JointDistribution method) (torch_mist.distributions.joint.categorical.JointCategorical method) (torch_mist.distributions.joint.wrapper.TorchJointDistribution method) (torch_mist.distributions.JointDistribution method) (torch_mist.estimators.generative.JointDistribution method) (torch_mist.estimators.JointDistribution method) _make_indexing_mask() (in module torch_mist.utils.indexing) _make_off_selection_indexing_mask() (in module torch_mist.utils.indexing) _marginal() (torch_mist.data.multimixture.MultivariateCorrelatedNormalMixture method) (torch_mist.data.multivariate.JointMultivariateNormal method) (torch_mist.distributions.joint.base.JointDistribution method) (torch_mist.distributions.joint.categorical.JointCategorical method) (torch_mist.distributions.JointDistribution method) (torch_mist.estimators.generative.JointDistribution method) (torch_mist.estimators.JointDistribution method) _mutual_information() (torch_mist.data.multimixture.MultivariateCorrelatedNormalMixture method) (torch_mist.distributions.joint.base.JointDistribution method) (torch_mist.distributions.JointDistribution method) (torch_mist.estimators.generative.JointDistribution method) (torch_mist.estimators.JointDistribution method) _params() (torch_mist.distributions.transforms.ConditionalLinear method), [1] (torch_mist.distributions.transforms.implementations.ConditionalLinear method) (torch_mist.distributions.transforms.implementations.Linear method) (torch_mist.distributions.transforms.implementations.linear.ConditionalLinear method) (torch_mist.distributions.transforms.implementations.linear.Linear method) (torch_mist.distributions.transforms.Linear method), [1] _prepare_k_fold_data() (in module torch_mist.utils.estimation) _reset_log() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.DummyLogger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) (torch_mist.utils.logging.logger.pandas.PandasLogger method) (torch_mist.utils.logging.logger.PandasLogger method) (torch_mist.utils.logging.logger.wandb.WandbLogger method) (torch_mist.utils.logging.PandasLogger method) _save_log() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) (torch_mist.utils.logging.logger.pandas.PandasLogger method) (torch_mist.utils.logging.logger.PandasLogger method) (torch_mist.utils.logging.PandasLogger method) _tensor_to_dict() (torch_mist.distributions.joint.categorical.JointCategorical method) (torch_mist.distributions.joint.wrapper.TorchJointDistribution method) _train_model() (torch_mist.decomposition.base.DimensionalityReduction method) (torch_mist.decomposition.CEB method) (torch_mist.decomposition.ceb.CEB method) (torch_mist.decomposition.MIB method) (torch_mist.decomposition.mib.MIB method) (torch_mist.decomposition.MID method) (torch_mist.decomposition.mid.MID method) (torch_mist.decomposition.VIB method) (torch_mist.decomposition.vib.VIB method) _train_on_fold() (in module torch_mist.utils.estimation) _unfold_variables() (torch_mist.estimators.transformed.base.TransformedMIEstimator method) (torch_mist.estimators.transformed.TransformedMIEstimator method) (torch_mist.estimators.TransformedMIEstimator method) _update_params() (torch_mist.distributions.transforms.EMANormalize method), [1] (torch_mist.distributions.transforms.implementations.EMANormalize method) (torch_mist.distributions.transforms.implementations.normalize.EMANormalize method) _update_vectors() (torch_mist.models.vqvae.VQVAE method) (torch_mist.quantization.VQVAE method)

A

add_cache() (in module torch_mist.distributions.caching) add_config() (torch_mist.utils.logging.logger.wandb.WandbLogger method) add_logging_hook() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) add_samples() (torch_mist.distributions.empirical.EmpiricalDistribution method) (torch_mist.distributions.EmpiricalDistribution method) alpha_tuba() (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) (in module torch_mist.estimators.discriminative.factories) AlphaTUBA (class in torch_mist.estimators), [1] (class in torch_mist.estimators.discriminative), [1] (class in torch_mist.estimators.discriminative.implementations) (class in torch_mist.estimators.discriminative.implementations.alpha_tuba) approx_log_p_x() (torch_mist.estimators.generative.base.JointGenerativeMIEstimator method) approx_log_p_xy() (torch_mist.estimators.generative.base.JointGenerativeMIEstimator method) approx_log_p_y() (torch_mist.estimators.CLUB method), [1] (torch_mist.estimators.DoE method), [1] (torch_mist.estimators.generative.base.GenerativeMIEstimator method) (torch_mist.estimators.generative.base.JointGenerativeMIEstimator method) (torch_mist.estimators.generative.CLUB method), [1] (torch_mist.estimators.generative.DoE method), [1] (torch_mist.estimators.generative.GenerativeMIEstimator method) (torch_mist.estimators.generative.implementations.CLUB method) (torch_mist.estimators.generative.implementations.club.CLUB method) (torch_mist.estimators.generative.implementations.DoE method) (torch_mist.estimators.generative.implementations.doe.DoE method) (torch_mist.estimators.generative.implementations.L1Out method) (torch_mist.estimators.generative.implementations.l1out.L1Out method) (torch_mist.estimators.generative.L1Out method), [1] (torch_mist.estimators.GenerativeMIEstimator method) (torch_mist.estimators.hybrid.GenerativeMIEstimator method) (torch_mist.estimators.L1Out method), [1]

approx_log_p_y_given_x() (torch_mist.estimators.generative.base.ConditionalGenerativeMIEstimator method) (torch_mist.estimators.generative.base.GenerativeMIEstimator method) (torch_mist.estimators.generative.base.JointGenerativeMIEstimator method) (torch_mist.estimators.generative.ConditionalGenerativeMIEstimator method) (torch_mist.estimators.generative.GenerativeMIEstimator method) (torch_mist.estimators.GenerativeMIEstimator method) (torch_mist.estimators.hybrid.GenerativeMIEstimator method) approx_log_partition() (torch_mist.estimators.discriminative.base.DiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.DiscriminativeMIEstimator method) (torch_mist.estimators.DiscriminativeMIEstimator method) (torch_mist.estimators.hybrid.DiscriminativeMIEstimator method), [1] ASYMMETRIC_HEADS (in module torch_mist.critic.factories)

B

BA (class in torch_mist.estimators), [1] (class in torch_mist.estimators.generative), [1] (class in torch_mist.estimators.generative.implementations) (class in torch_mist.estimators.generative.implementations.ba) ba() (in module torch_mist.estimators) (in module torch_mist.estimators.generative) (in module torch_mist.estimators.generative.factories) Baseline (class in torch_mist.baseline) (class in torch_mist.baseline.base) baseline_nn() (in module torch_mist.baseline) (in module torch_mist.baseline.factories) (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) BaselineDiscriminativeMIEstimator (class in torch_mist.estimators.discriminative) (class in torch_mist.estimators.discriminative.base) batch_loss() (torch_mist.estimators.base.MIEstimator method) (torch_mist.estimators.discriminative.base.DiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.DiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.FLO method), [1] (torch_mist.estimators.discriminative.implementations.FLO method) (torch_mist.estimators.discriminative.implementations.flo.FLO method) (torch_mist.estimators.discriminative.implementations.JS method) (torch_mist.estimators.discriminative.implementations.js.JS method) (torch_mist.estimators.discriminative.implementations.MINE method) (torch_mist.estimators.discriminative.implementations.mine.MINE method) (torch_mist.estimators.discriminative.implementations.SMILE method) (torch_mist.estimators.discriminative.implementations.smile.SMILE method) (torch_mist.estimators.discriminative.JS method), [1] (torch_mist.estimators.discriminative.MINE method), [1] (torch_mist.estimators.discriminative.SMILE method), [1] (torch_mist.estimators.DiscriminativeMIEstimator method) (torch_mist.estimators.DoE method), [1] (torch_mist.estimators.DummyGenerativeMIEstimator method), [1] (torch_mist.estimators.FLO method), [1] (torch_mist.estimators.generative.base.ConditionalGenerativeMIEstimator method) (torch_mist.estimators.generative.base.JointGenerativeMIEstimator method) (torch_mist.estimators.generative.ConditionalGenerativeMIEstimator method) (torch_mist.estimators.generative.DoE method), [1] (torch_mist.estimators.generative.DummyGenerativeMIEstimator method), [1] (torch_mist.estimators.generative.GM method), [1] (torch_mist.estimators.generative.implementations.DoE method) (torch_mist.estimators.generative.implementations.doe.DoE method) (torch_mist.estimators.generative.implementations.dummy.DummyGenerativeMIEstimator method) (torch_mist.estimators.generative.implementations.DummyGenerativeMIEstimator method) (torch_mist.estimators.generative.implementations.GM method) (torch_mist.estimators.generative.implementations.gm.GM method) (torch_mist.estimators.GM method), [1] (torch_mist.estimators.hybrid.base.HybridMIEstimator method) (torch_mist.estimators.hybrid.DiscriminativeMIEstimator method), [1] (torch_mist.estimators.hybrid.HybridMIEstimator method) (torch_mist.estimators.hybrid.MIEstimator method), [1] (torch_mist.estimators.HybridMIEstimator method) (torch_mist.estimators.JS method), [1] (torch_mist.estimators.MIEstimator method), [1], [2] (torch_mist.estimators.MINE method), [1] (torch_mist.estimators.multi.MultiMIEstimator method) (torch_mist.estimators.MultiMIEstimator method) (torch_mist.estimators.SMILE method), [1] (torch_mist.estimators.transformed.base.TransformedMIEstimator method) (torch_mist.estimators.transformed.implementations.flip.FlippedMIEstimator method) (torch_mist.estimators.transformed.TransformedMIEstimator method) (torch_mist.estimators.TransformedMIEstimator method)

BatchLogMeanExp (class in torch_mist.baseline) (class in torch_mist.baseline.base) bijective (torch_mist.distributions.transforms.ConditionalLinear attribute), [1] (torch_mist.distributions.transforms.implementations.ConditionalLinear attribute) (torch_mist.distributions.transforms.implementations.linear.ConditionalLinear attribute) (torch_mist.distributions.transforms.implementations.linear.ConditionedLinear attribute) binned() (in module torch_mist.estimators) (in module torch_mist.estimators.transformed) (in module torch_mist.estimators.transformed.factories) BinnedMIEstimator (class in torch_mist.estimators), [1] (class in torch_mist.estimators.transformed), [1] (class in torch_mist.estimators.transformed.implementations) (class in torch_mist.estimators.transformed.implementations.binned) broadcast_function() (torch_mist.estimators.multi.MultiMIEstimator method) (torch_mist.estimators.MultiMIEstimator method)

C

cached_function() (in module torch_mist.utils.caching) cached_method() (in module torch_mist.estimators) (in module torch_mist.estimators.hybrid) (in module torch_mist.utils.caching) CachedConditionalTransformModule (class in torch_mist.distributions.caching) CachedModule (class in torch_mist.nn) CachedTransformModule (class in torch_mist.distributions.caching) categorical (torch_mist.distributions.joint.categorical.JointCategorical property) CategoricalModule (class in torch_mist.distributions) (class in torch_mist.distributions.categorical) CEB (class in torch_mist.decomposition) (class in torch_mist.decomposition.ceb) (class in torch_mist.models) (class in torch_mist.models.bottleneck) CenterAndScale (class in torch_mist.decomposition.base) clear() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) clear_cache() (torch_mist.distributions.transforms.base.ConditionalTransformedDistributionModule method) (torch_mist.distributions.transforms.ConditionalTransformedDistributionModule method) CLUB (class in torch_mist.estimators), [1] (class in torch_mist.estimators.generative), [1] (class in torch_mist.estimators.generative.implementations) (class in torch_mist.estimators.generative.implementations.club) club() (in module torch_mist.estimators) (in module torch_mist.estimators.generative) (in module torch_mist.estimators.generative.factories) ClusterQuantization (class in torch_mist.quantization) (class in torch_mist.quantization.functions) codebook_lookup() (torch_mist.quantization.functions.VectorQuantization method) (torch_mist.quantization.VectorQuantization method) codomain (torch_mist.distributions.transforms.ConditionalLinear attribute), [1] (torch_mist.distributions.transforms.implementations.ConditionalLinear attribute) (torch_mist.distributions.transforms.implementations.linear.ConditionalLinear attribute) (torch_mist.distributions.transforms.implementations.linear.ConditionedLinear attribute) compute_mean() (in module torch_mist.utils.logging.metrics) compute_mean_std() (in module torch_mist.utils.logging.metrics) compute_mi() (in module torch_mist.data.utils) compute_training_time() (in module torch_mist.utils.train.model) condition() (torch_mist.distributions.caching.CachedConditionalTransformModule method) (torch_mist.distributions.categorical.ConditionalCategoricalModule method) (torch_mist.distributions.conditional.delta.ConditionalDelta method) (torch_mist.distributions.conditional.unconditional.UnconditionalDistribution method) (torch_mist.distributions.ConditionalCategoricalModule method), [1] (torch_mist.distributions.joint.base.ConditionalNamedDistribution method) (torch_mist.distributions.joint.base.JointDistribution method) (torch_mist.distributions.JointDistribution method) (torch_mist.distributions.normal.ConditionalStandardNormalModule method) (torch_mist.distributions.transforms.base.ConditionalTransformedDistributionModule method) (torch_mist.distributions.transforms.ConditionalLinear method), [1] (torch_mist.distributions.transforms.ConditionalTransformedDistributionModule method) (torch_mist.distributions.transforms.implementations.ConditionalLinear method) (torch_mist.distributions.transforms.implementations.linear.ConditionalLinear method) (torch_mist.estimators.generative.JointDistribution method) (torch_mist.estimators.JointDistribution method)

conditional() (torch_mist.distributions.joint.base.JointDistribution method) (torch_mist.distributions.JointDistribution method) (torch_mist.estimators.generative.JointDistribution method) (torch_mist.estimators.JointDistribution method) conditional_categorical() (in module torch_mist.distributions) (in module torch_mist.distributions.factories) (in module torch_mist.estimators) (in module torch_mist.estimators.hybrid) conditional_linear() (in module torch_mist.distributions.transforms) (in module torch_mist.distributions.transforms.factories) conditional_transformed_normal() (in module torch_mist.distributions) (in module torch_mist.distributions.factories) (in module torch_mist.quantization) ConditionalCategoricalModule (class in torch_mist.distributions), [1] (class in torch_mist.distributions.categorical) ConditionalDelta (class in torch_mist.distributions.conditional.delta) ConditionalDistributionModule (class in torch_mist.distributions) (class in torch_mist.distributions.transforms) (class in torch_mist.distributions.transforms.base) ConditionalGenerativeMIEstimator (class in torch_mist.estimators.generative) (class in torch_mist.estimators.generative.base) ConditionalLinear (class in torch_mist.distributions.transforms), [1] (class in torch_mist.distributions.transforms.implementations) (class in torch_mist.distributions.transforms.implementations.linear) ConditionalNamedDistribution (class in torch_mist.distributions.joint.base) ConditionalStandardNormalModule (class in torch_mist.distributions.normal) ConditionalTransformedDistributionModule (class in torch_mist.distributions.transforms) (class in torch_mist.distributions.transforms.base) ConditionalTransformedNormalModule (class in torch_mist.distributions), [1] (class in torch_mist.distributions.normal) ConditionedJointDistribution (class in torch_mist.distributions.joint.base) ConditionedLinear (class in torch_mist.distributions.transforms.implementations.linear) ConstantBaseline (class in torch_mist.baseline) (class in torch_mist.baseline.base) (class in torch_mist.estimators) (class in torch_mist.estimators.discriminative) ConstantCritic (class in torch_mist.critic.constant) convert_to_tensor() (in module torch_mist.utils.data.utils) Critic (class in torch_mist.critic) (class in torch_mist.critic.base) critic_nn() (in module torch_mist.critic) (in module torch_mist.critic.factories) (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) CRITIC_TYPE (in module torch_mist.critic.base) CRITIC_TYPES (in module torch_mist.critic.base) CSVDataset (class in torch_mist.utils.data) (class in torch_mist.utils.data.csv)

D

DataFrameDataset (class in torch_mist.utils.data.dataset) DEFAULT_BATCH_SIZE (in module torch_mist.decomposition.base) (in module torch_mist.decomposition.ceb) (in module torch_mist.decomposition.mib) (in module torch_mist.decomposition.mid) (in module torch_mist.decomposition.tib) (in module torch_mist.decomposition.tinfomax) (in module torch_mist.decomposition.vib) DEFAULT_HIDDEN_DIMS (in module torch_mist.estimators) (in module torch_mist.estimators.factories) DEFAULT_MAX_ITERATIONS (in module torch_mist.decomposition.base) (in module torch_mist.decomposition.ceb) (in module torch_mist.decomposition.mib) (in module torch_mist.decomposition.mid) (in module torch_mist.decomposition.tib) (in module torch_mist.decomposition.tinfomax) (in module torch_mist.decomposition.vib) DEFAULTS (in module torch_mist.utils.estimation) delete_best_weights() (torch_mist.utils.train.utils.RunTerminationManager method) delete_cache_hook() (in module torch_mist.utils.caching) delete_unused_kwargs() (in module torch_mist.distributions) (in module torch_mist.distributions.factories) DeltaMap (class in torch_mist.distributions.parametrizations.map) dense_nn() (in module torch_mist.nn) (in module torch_mist.quantization) detach_all_hooks() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) detach_hook() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) DimensionalityReduction (class in torch_mist.decomposition.base) disable_batch_validation() (torch_mist.estimators.hybrid.implementations.hybrid_pq.PQHybridMIEstimator method) (torch_mist.estimators.hybrid.implementations.PQHybridMIEstimator method) (torch_mist.estimators.hybrid.PQHybridMIEstimator method) (torch_mist.estimators.PQHybridMIEstimator method)

DiscriminativeMIEstimator (class in torch_mist.estimators) (class in torch_mist.estimators.discriminative) (class in torch_mist.estimators.discriminative.base) (class in torch_mist.estimators.hybrid), [1] DistributionDataset (class in torch_mist.utils.data.dataset) DistributionModule (class in torch_mist.distributions.transforms) (class in torch_mist.distributions.transforms.base) DoE (class in torch_mist.estimators), [1] (class in torch_mist.estimators.generative), [1] (class in torch_mist.estimators.generative.implementations) (class in torch_mist.estimators.generative.implementations.doe) doe() (in module torch_mist.estimators) (in module torch_mist.estimators.generative) (in module torch_mist.estimators.generative.factories) domain (torch_mist.distributions.transforms.ConditionalLinear attribute), [1] (torch_mist.distributions.transforms.implementations.ConditionalLinear attribute) (torch_mist.distributions.transforms.implementations.linear.ConditionalLinear attribute) (torch_mist.distributions.transforms.implementations.linear.ConditionedLinear attribute) dummy_discriminative() (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) (in module torch_mist.estimators.discriminative.factories) dummy_generative() (in module torch_mist.estimators) (in module torch_mist.estimators.generative) (in module torch_mist.estimators.generative.factories) DummyDiscriminativeMIEstimator (class in torch_mist.estimators), [1] (class in torch_mist.estimators.discriminative), [1] (class in torch_mist.estimators.discriminative.implementations) (class in torch_mist.estimators.discriminative.implementations.dummy) DummyGenerativeMIEstimator (class in torch_mist.estimators), [1] (class in torch_mist.estimators.generative), [1] (class in torch_mist.estimators.generative.implementations) (class in torch_mist.estimators.generative.implementations.dummy) DummyLogger (class in torch_mist.utils.logging.logger.base) DummyModule (class in torch_mist.estimators.transformed.utils)

E

emanorm() (in module torch_mist.distributions.transforms) (in module torch_mist.distributions.transforms.factories) EMANormalize (class in torch_mist.distributions.transforms), [1] (class in torch_mist.distributions.transforms.implementations) (class in torch_mist.distributions.transforms.implementations.normalize) EmpiricalDistribution (class in torch_mist.distributions) (class in torch_mist.distributions.empirical) enable_batch_validation() (torch_mist.estimators.hybrid.implementations.hybrid_pq.PQHybridMIEstimator method) (torch_mist.estimators.hybrid.implementations.PQHybridMIEstimator method) (torch_mist.estimators.hybrid.PQHybridMIEstimator method) (torch_mist.estimators.PQHybridMIEstimator method) entropy() (torch_mist.distributions.joint.base.JointDistribution method) (torch_mist.distributions.JointDistribution method) (torch_mist.estimators.generative.JointDistribution method) (torch_mist.estimators.JointDistribution method) epoch() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method)

EPSILON (in module torch_mist.distributions.parametrizations.map) ERROR_MESSAGE (in module torch_mist.estimators.transformed.base) estimate_mi() (in module torch_mist) (in module torch_mist.utils) (in module torch_mist.utils.estimation) estimate_temporal_mi() (in module torch_mist.utils.estimation) evaluate() (in module torch_mist.utils.evaluation) (in module torch_mist.utils.evaluation.model) evaluate_mi() (in module torch_mist.utils) (in module torch_mist.utils.evaluation) (in module torch_mist.utils.evaluation.mi_estimator) expand_to_same_shape() (in module torch_mist.utils.shape) ExponentialMovingAverage (class in torch_mist.baseline) (class in torch_mist.baseline.base)

F

fetch_transform() (in module torch_mist.distributions) (in module torch_mist.distributions.transforms) (in module torch_mist.distributions.transforms.utils) filter_dataset() (in module torch_mist.utils.data.utils) fit() (torch_mist.decomposition.base.DimensionalityReduction method) (torch_mist.decomposition.TIB method) (torch_mist.decomposition.tib.TIB method) (torch_mist.decomposition.tinfomax.TInfoMax method) (torch_mist.quantization.functions.LearnableQuantization method) (torch_mist.quantization.LearnableQuantization method) FixedQuantization (class in torch_mist.quantization) (class in torch_mist.quantization.functions) flip_estimator() (in module torch_mist.estimators) (in module torch_mist.estimators.utils) FlippedMIEstimator (class in torch_mist.estimators) (class in torch_mist.estimators.transformed.implementations.flip) (class in torch_mist.estimators.utils) FLO (class in torch_mist.estimators), [1] (class in torch_mist.estimators.discriminative), [1] (class in torch_mist.estimators.discriminative.implementations) (class in torch_mist.estimators.discriminative.implementations.flo) flo() (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) (in module torch_mist.estimators.discriminative.factories) forward() (torch_mist.baseline.base.Baseline method) (torch_mist.baseline.base.BatchLogMeanExp method) (torch_mist.baseline.base.ConstantBaseline method) (torch_mist.baseline.base.ExponentialMovingAverage method) (torch_mist.baseline.base.InterpolatedBaseline method) (torch_mist.baseline.base.LearnableBaseline method) (torch_mist.baseline.Baseline method) (torch_mist.baseline.BatchLogMeanExp method) (torch_mist.baseline.ConstantBaseline method) (torch_mist.baseline.ExponentialMovingAverage method) (torch_mist.baseline.InterpolatedBaseline method) (torch_mist.baseline.LearnableBaseline method) (torch_mist.critic.base.Critic method) (torch_mist.critic.constant.ConstantCritic method) (torch_mist.critic.Critic method) (torch_mist.critic.joint.JointCritic method) (torch_mist.critic.JointCritic method) (torch_mist.critic.separable.SeparableCritic method) (torch_mist.critic.SeparableCritic method) (torch_mist.estimators.base.MIEstimator method) (torch_mist.estimators.ConstantBaseline method) (torch_mist.estimators.discriminative.ConstantBaseline method) (torch_mist.estimators.hybrid.MIEstimator method), [1] (torch_mist.estimators.hybrid.QuantizationFunction method) (torch_mist.estimators.MIEstimator method), [1], [2] (torch_mist.estimators.multi.MultiMIEstimator method) (torch_mist.estimators.MultiMIEstimator method) (torch_mist.estimators.QuantizationFunction method), [1] (torch_mist.estimators.transformed.base.TransformedMIEstimator method) (torch_mist.estimators.transformed.implementations.flip.FlippedMIEstimator method) (torch_mist.estimators.transformed.QuantizationFunction method) (torch_mist.estimators.transformed.TransformedMIEstimator method) (torch_mist.estimators.transformed.utils.DummyModule method) (torch_mist.estimators.TransformedMIEstimator method) (torch_mist.nn.CachedModule method) (torch_mist.nn.Identity method) (torch_mist.nn.Normalize method) (torch_mist.quantization.functions.LearnableQuantization method) (torch_mist.quantization.functions.QuantizationFunction method) (torch_mist.quantization.LearnableQuantization method) (torch_mist.quantization.QuantizationFunction method), [1]

freeze() (in module torch_mist.utils) (in module torch_mist.utils.freeze) full_name() (torch_mist.distributions.joint.base.JointDistribution method) (torch_mist.distributions.JointDistribution method) (torch_mist.estimators.generative.JointDistribution method) (torch_mist.estimators.JointDistribution method)

G

GenerativeMIEstimator (class in torch_mist.estimators) (class in torch_mist.estimators.generative) (class in torch_mist.estimators.generative.base) (class in torch_mist.estimators.hybrid) get_log() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) (torch_mist.utils.logging.logger.pandas.PandasLogger method) (torch_mist.utils.logging.logger.PandasLogger method) (torch_mist.utils.logging.logger.wandb.WandbLogger method) (torch_mist.utils.logging.PandasLogger method)

GM (class in torch_mist.estimators), [1] (class in torch_mist.estimators.generative), [1] (class in torch_mist.estimators.generative.implementations) (class in torch_mist.estimators.generative.implementations.gm) gm() (in module torch_mist.estimators) (in module torch_mist.estimators.generative) (in module torch_mist.estimators.generative.factories)

H

hybrid_pq() (in module torch_mist.estimators) (in module torch_mist.estimators.hybrid) (in module torch_mist.estimators.hybrid.factories)

HybridMIEstimator (class in torch_mist.estimators) (class in torch_mist.estimators.hybrid) (class in torch_mist.estimators.hybrid.base)

I

Identity (class in torch_mist.nn) infer_dims() (in module torch_mist.utils.data.utils) infomax_gradient (torch_mist.estimators.BA attribute), [1] (torch_mist.estimators.base.MIEstimator attribute) (torch_mist.estimators.CLUB attribute), [1] (torch_mist.estimators.discriminative.base.DiscriminativeMIEstimator attribute) (torch_mist.estimators.discriminative.DiscriminativeMIEstimator attribute) (torch_mist.estimators.DiscriminativeMIEstimator attribute) (torch_mist.estimators.DoE attribute), [1] (torch_mist.estimators.DummyGenerativeMIEstimator attribute), [1] (torch_mist.estimators.generative.BA attribute), [1] (torch_mist.estimators.generative.CLUB attribute), [1] (torch_mist.estimators.generative.DoE attribute), [1] (torch_mist.estimators.generative.DummyGenerativeMIEstimator attribute), [1] (torch_mist.estimators.generative.implementations.BA attribute) (torch_mist.estimators.generative.implementations.ba.BA attribute) (torch_mist.estimators.generative.implementations.CLUB attribute) (torch_mist.estimators.generative.implementations.club.CLUB attribute) (torch_mist.estimators.generative.implementations.DoE attribute) (torch_mist.estimators.generative.implementations.doe.DoE attribute) (torch_mist.estimators.generative.implementations.dummy.DummyGenerativeMIEstimator attribute) (torch_mist.estimators.generative.implementations.DummyGenerativeMIEstimator attribute) (torch_mist.estimators.generative.implementations.L1Out attribute) (torch_mist.estimators.generative.implementations.l1out.L1Out attribute) (torch_mist.estimators.generative.L1Out attribute), [1] (torch_mist.estimators.hybrid.DiscriminativeMIEstimator attribute), [1] (torch_mist.estimators.hybrid.MIEstimator attribute), [1] (torch_mist.estimators.L1Out attribute), [1] (torch_mist.estimators.MIEstimator attribute), [1], [2] InfoNCE (class in torch_mist.estimators), [1] (class in torch_mist.estimators.discriminative), [1] (class in torch_mist.estimators.discriminative.implementations) (class in torch_mist.estimators.discriminative.implementations.infonce)

infonce() (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) (in module torch_mist.estimators.discriminative.factories) InformationBottleneck (class in torch_mist.models.bottleneck) instantiate_estimator() (in module torch_mist.estimators), [1] (in module torch_mist.estimators.factories) (in module torch_mist.estimators.hybrid) instantiate_optimizer() (in module torch_mist.utils.train.model) instantiate_quantization() (in module torch_mist.estimators), [1] (in module torch_mist.estimators.hybrid) (in module torch_mist.estimators.transformed) (in module torch_mist.quantization) (in module torch_mist.quantization.factories) InterpolatedBaseline (class in torch_mist.baseline) (class in torch_mist.baseline.base) is_data_loader() (in module torch_mist.utils.data.utils) is_early_stopping_possible() (in module torch_mist.utils.train.model) is_logged() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) is_trainable() (in module torch_mist.estimators) (in module torch_mist.estimators.hybrid) (in module torch_mist.utils) (in module torch_mist.utils.freeze) is_valid_entry() (in module torch_mist.utils.data.utils) iteration() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method)

J

JOINT_CRITIC (in module torch_mist.critic.base) (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) joint_critic() (in module torch_mist.critic) (in module torch_mist.critic.factories) joint_transformed_normal() (in module torch_mist.distributions) (in module torch_mist.distributions.factories) (in module torch_mist.estimators) (in module torch_mist.estimators.generative) JointCategorical (class in torch_mist.distributions.joint.categorical) JointCritic (class in torch_mist.critic) (class in torch_mist.critic.joint) JointDistribution (class in torch_mist.distributions) (class in torch_mist.distributions.joint.base) (class in torch_mist.estimators) (class in torch_mist.estimators.generative)

JointGenerativeMIEstimator (class in torch_mist.estimators.generative.base) JointMultivariateNormal (class in torch_mist.data.multivariate) JointTransformedNormalModule (class in torch_mist.distributions), [1] (class in torch_mist.distributions.normal) JS (class in torch_mist.estimators), [1] (class in torch_mist.estimators.discriminative), [1] (class in torch_mist.estimators.discriminative.implementations) (class in torch_mist.estimators.discriminative.implementations.js) js() (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) (in module torch_mist.estimators.discriminative.factories)

K

k_fold_mi_estimate() (in module torch_mist) (in module torch_mist.utils) (in module torch_mist.utils.estimation)

kmeans_quantization() (in module torch_mist.quantization) (in module torch_mist.quantization.factories)

L

L1Out (class in torch_mist.estimators), [1] (class in torch_mist.estimators.generative), [1] (class in torch_mist.estimators.generative.implementations) (class in torch_mist.estimators.generative.implementations.l1out) l1out() (in module torch_mist.estimators) (in module torch_mist.estimators.generative) (in module torch_mist.estimators.generative.factories) LearnableBaseline (class in torch_mist.baseline) (class in torch_mist.baseline.base) LearnableQuantization (class in torch_mist.quantization) (class in torch_mist.quantization.functions) LearnableVectorQuantization (class in torch_mist.quantization) (class in torch_mist.quantization.functions) Linear (class in torch_mist.distributions.transforms), [1] (class in torch_mist.distributions.transforms.implementations) (class in torch_mist.distributions.transforms.implementations.linear) linear() (in module torch_mist.distributions.transforms) (in module torch_mist.distributions.transforms.factories) load_best_weights() (torch_mist.utils.train.utils.RunTerminationManager method) LocScaleMap (class in torch_mist.distributions.parametrizations.map) log() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) log_abs_det_jacobian() (torch_mist.distributions.caching.CachedTransformModule method) (torch_mist.distributions.transforms.implementations.linear.ConditionedLinear method) (torch_mist.distributions.transforms.implementations.Permute method) (torch_mist.distributions.transforms.implementations.permute.Permute method) (torch_mist.distributions.transforms.Permute method), [1] log_prob() (torch_mist.distributions.categorical.CategoricalModule method) (torch_mist.distributions.CategoricalModule method) (torch_mist.distributions.joint.base.JointDistribution method) (torch_mist.distributions.JointDistribution method) (torch_mist.distributions.normal.NormalModule method) (torch_mist.distributions.NormalModule method) (torch_mist.distributions.transforms.base.TransformedDistributionModule method) (torch_mist.distributions.transforms.TransformedDistributionModule method) (torch_mist.estimators.generative.JointDistribution method) (torch_mist.estimators.JointDistribution method) log_ratio() (torch_mist.estimators.base.MIEstimator method) (torch_mist.estimators.discriminative.base.DiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.DiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.FLO method), [1] (torch_mist.estimators.discriminative.implementations.FLO method) (torch_mist.estimators.discriminative.implementations.flo.FLO method) (torch_mist.estimators.discriminative.implementations.SMILE method) (torch_mist.estimators.discriminative.implementations.smile.SMILE method) (torch_mist.estimators.discriminative.SMILE method), [1] (torch_mist.estimators.DiscriminativeMIEstimator method) (torch_mist.estimators.DummyGenerativeMIEstimator method), [1] (torch_mist.estimators.FlippedMIEstimator method) (torch_mist.estimators.FLO method), [1] (torch_mist.estimators.generative.base.GenerativeMIEstimator method) (torch_mist.estimators.generative.DummyGenerativeMIEstimator method), [1] (torch_mist.estimators.generative.GenerativeMIEstimator method) (torch_mist.estimators.generative.implementations.dummy.DummyGenerativeMIEstimator method) (torch_mist.estimators.generative.implementations.DummyGenerativeMIEstimator method) (torch_mist.estimators.GenerativeMIEstimator method) (torch_mist.estimators.hybrid.base.HybridMIEstimator method) (torch_mist.estimators.hybrid.DiscriminativeMIEstimator method), [1] (torch_mist.estimators.hybrid.GenerativeMIEstimator method) (torch_mist.estimators.hybrid.HybridMIEstimator method) (torch_mist.estimators.hybrid.MIEstimator method), [1] (torch_mist.estimators.HybridMIEstimator method) (torch_mist.estimators.MIEstimator method), [1], [2] (torch_mist.estimators.multi.MultiMIEstimator method) (torch_mist.estimators.MultiMIEstimator method) (torch_mist.estimators.SMILE method), [1] (torch_mist.estimators.transformed.base.TransformedMIEstimator method) (torch_mist.estimators.transformed.implementations.flip.FlippedMIEstimator method) (torch_mist.estimators.transformed.TransformedMIEstimator method) (torch_mist.estimators.TransformedMIEstimator method) (torch_mist.estimators.utils.FlippedMIEstimator method)

logged_methods() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) Logger (class in torch_mist.utils.logging) (class in torch_mist.utils.logging.logger) (class in torch_mist.utils.logging.logger.base) LogitsMap (class in torch_mist.distributions.parametrizations.map) loss() (torch_mist.estimators.base.MIEstimator method) (torch_mist.estimators.FlippedMIEstimator method) (torch_mist.estimators.hybrid.MIEstimator method), [1] (torch_mist.estimators.MIEstimator method), [1], [2] (torch_mist.estimators.multi.MultiMIEstimator method) (torch_mist.estimators.MultiMIEstimator method) (torch_mist.estimators.transformed.base.TransformedMIEstimator method) (torch_mist.estimators.transformed.implementations.flip.FlippedMIEstimator method) (torch_mist.estimators.transformed.TransformedMIEstimator method) (torch_mist.estimators.TransformedMIEstimator method) (torch_mist.estimators.utils.FlippedMIEstimator method) (torch_mist.models.bottleneck.InformationBottleneck method) (torch_mist.models.vqvae.VQVAE method) (torch_mist.nn.Model method) (torch_mist.quantization.functions.LearnableQuantization method) (torch_mist.quantization.LearnableQuantization method) (torch_mist.quantization.VQVAE method) lower_bound (torch_mist.estimators.BA attribute), [1] (torch_mist.estimators.BinnedMIEstimator attribute), [1] (torch_mist.estimators.discriminative.base.DiscriminativeMIEstimator attribute) (torch_mist.estimators.discriminative.DiscriminativeMIEstimator attribute) (torch_mist.estimators.discriminative.implementations.MINE attribute) (torch_mist.estimators.discriminative.implementations.mine.MINE attribute) (torch_mist.estimators.discriminative.MINE attribute), [1] (torch_mist.estimators.DiscriminativeMIEstimator attribute) (torch_mist.estimators.DummyGenerativeMIEstimator attribute), [1] (torch_mist.estimators.generative.BA attribute), [1] (torch_mist.estimators.generative.DummyGenerativeMIEstimator attribute), [1] (torch_mist.estimators.generative.implementations.BA attribute) (torch_mist.estimators.generative.implementations.ba.BA attribute) (torch_mist.estimators.generative.implementations.dummy.DummyGenerativeMIEstimator attribute) (torch_mist.estimators.generative.implementations.DummyGenerativeMIEstimator attribute) (torch_mist.estimators.hybrid.DiscriminativeMIEstimator attribute), [1] (torch_mist.estimators.MINE attribute), [1] (torch_mist.estimators.transformed.BinnedMIEstimator attribute), [1] (torch_mist.estimators.transformed.implementations.binned.BinnedMIEstimator attribute) (torch_mist.estimators.transformed.implementations.BinnedMIEstimator attribute) (torch_mist.nn.Model attribute)

M

make_dataloader() (in module torch_mist.utils.data.utils) make_dataset() (in module torch_mist.utils.data.utils) make_default_dataloader() (in module torch_mist.utils.data.utils) make_default_dataloaders() (in module torch_mist.utils.data.utils) make_splits() (in module torch_mist.utils.data.utils) make_transforms() (in module torch_mist.distributions) (in module torch_mist.distributions.factories) map_parameters() (torch_mist.distributions.parametrizations.map.DeltaMap method) (torch_mist.distributions.parametrizations.map.LocScaleMap method) (torch_mist.distributions.parametrizations.map.LogitsMap method) (torch_mist.distributions.parametrizations.map.ParameterMap method) marginal() (torch_mist.distributions.joint.base.JointDistribution method) (torch_mist.distributions.JointDistribution method) (torch_mist.estimators.generative.JointDistribution method) (torch_mist.estimators.JointDistribution method) matrix_off_diagonal() (in module torch_mist.utils.indexing) MC_ITERATIONS (torch_mist.data.multimixture.MultivariateCorrelatedNormalMixture attribute) MC_SAMPLES (torch_mist.data.multimixture.MultivariateCorrelatedNormalMixture attribute) MIB (class in torch_mist.decomposition) (class in torch_mist.decomposition.mib) (class in torch_mist.models) (class in torch_mist.models.bottleneck) MID (class in torch_mist.decomposition) (class in torch_mist.decomposition.mid) MIEstimator (class in torch_mist.estimators), [1], [2] (class in torch_mist.estimators.base) (class in torch_mist.estimators.hybrid), [1] MINE (class in torch_mist.estimators), [1] (class in torch_mist.estimators.discriminative), [1] (class in torch_mist.estimators.discriminative.implementations) (class in torch_mist.estimators.discriminative.implementations.mine) mine() (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) (in module torch_mist.estimators.discriminative.factories) Model (class in torch_mist.nn) module torch_mist torch_mist.baseline torch_mist.baseline.base torch_mist.baseline.factories torch_mist.critic torch_mist.critic.base torch_mist.critic.constant torch_mist.critic.factories torch_mist.critic.joint torch_mist.critic.separable torch_mist.data torch_mist.data.multimixture torch_mist.data.multivariate torch_mist.data.utils torch_mist.decomposition torch_mist.decomposition.base torch_mist.decomposition.ceb torch_mist.decomposition.mib torch_mist.decomposition.mid torch_mist.decomposition.tib torch_mist.decomposition.tinfomax torch_mist.decomposition.vib torch_mist.distributions torch_mist.distributions.caching torch_mist.distributions.categorical torch_mist.distributions.conditional torch_mist.distributions.conditional.delta torch_mist.distributions.conditional.unconditional torch_mist.distributions.empirical torch_mist.distributions.factories torch_mist.distributions.joint torch_mist.distributions.joint.base torch_mist.distributions.joint.categorical torch_mist.distributions.joint.wrapper torch_mist.distributions.normal torch_mist.distributions.parametrizations torch_mist.distributions.parametrizations.map torch_mist.distributions.transforms torch_mist.distributions.transforms.base torch_mist.distributions.transforms.factories torch_mist.distributions.transforms.implementations torch_mist.distributions.transforms.implementations.linear torch_mist.distributions.transforms.implementations.normalize torch_mist.distributions.transforms.implementations.permute torch_mist.distributions.transforms.utils torch_mist.estimators torch_mist.estimators.base torch_mist.estimators.discriminative torch_mist.estimators.discriminative.base torch_mist.estimators.discriminative.factories torch_mist.estimators.discriminative.implementations torch_mist.estimators.discriminative.implementations.alpha_tuba torch_mist.estimators.discriminative.implementations.dummy torch_mist.estimators.discriminative.implementations.flo torch_mist.estimators.discriminative.implementations.infonce torch_mist.estimators.discriminative.implementations.js torch_mist.estimators.discriminative.implementations.mine torch_mist.estimators.discriminative.implementations.nwj torch_mist.estimators.discriminative.implementations.smile torch_mist.estimators.discriminative.implementations.tuba torch_mist.estimators.factories torch_mist.estimators.generative torch_mist.estimators.generative.base torch_mist.estimators.generative.factories torch_mist.estimators.generative.implementations torch_mist.estimators.generative.implementations.ba torch_mist.estimators.generative.implementations.club torch_mist.estimators.generative.implementations.doe torch_mist.estimators.generative.implementations.dummy torch_mist.estimators.generative.implementations.gm torch_mist.estimators.generative.implementations.l1out torch_mist.estimators.hybrid torch_mist.estimators.hybrid.base torch_mist.estimators.hybrid.factories torch_mist.estimators.hybrid.implementations torch_mist.estimators.hybrid.implementations.hybrid_pq torch_mist.estimators.hybrid.implementations.resampled torch_mist.estimators.hybrid.implementations.reweighted torch_mist.estimators.multi torch_mist.estimators.temporal torch_mist.estimators.transformed torch_mist.estimators.transformed.base torch_mist.estimators.transformed.factories torch_mist.estimators.transformed.implementations torch_mist.estimators.transformed.implementations.binned torch_mist.estimators.transformed.implementations.flip torch_mist.estimators.transformed.implementations.pq torch_mist.estimators.transformed.utils torch_mist.estimators.utils torch_mist.models torch_mist.models.bottleneck torch_mist.models.vqvae torch_mist.nn torch_mist.quantization torch_mist.quantization.factories torch_mist.quantization.functions torch_mist.utils torch_mist.utils.caching torch_mist.utils.data torch_mist.utils.data.csv torch_mist.utils.data.dataset torch_mist.utils.data.loader torch_mist.utils.data.sampler torch_mist.utils.data.temporal torch_mist.utils.data.utils torch_mist.utils.estimation torch_mist.utils.evaluation torch_mist.utils.evaluation.mi_estimator torch_mist.utils.evaluation.model torch_mist.utils.freeze torch_mist.utils.indexing torch_mist.utils.logging torch_mist.utils.logging.logger torch_mist.utils.logging.logger.base torch_mist.utils.logging.logger.pandas torch_mist.utils.logging.logger.wandb torch_mist.utils.logging.metrics torch_mist.utils.shape torch_mist.utils.train torch_mist.utils.train.mi_estimator torch_mist.utils.train.model torch_mist.utils.train.utils

multi_head_dense_nn() (in module torch_mist.nn) MultiMIEstimator (class in torch_mist.estimators) (class in torch_mist.estimators.multi) MultivariateCorrelatedNormalMixture (class in torch_mist.data.multimixture) mutual_information() (torch_mist.distributions.joint.base.JointDistribution method) (torch_mist.distributions.JointDistribution method) (torch_mist.estimators.BA method), [1] (torch_mist.estimators.base.MIEstimator method) (torch_mist.estimators.FlippedMIEstimator method) (torch_mist.estimators.generative.BA method), [1] (torch_mist.estimators.generative.implementations.BA method) (torch_mist.estimators.generative.implementations.ba.BA method) (torch_mist.estimators.generative.JointDistribution method) (torch_mist.estimators.hybrid.MIEstimator method), [1] (torch_mist.estimators.JointDistribution method) (torch_mist.estimators.MIEstimator method), [1], [2] (torch_mist.estimators.multi.MultiMIEstimator method) (torch_mist.estimators.MultiMIEstimator method) (torch_mist.estimators.transformed.base.TransformedMIEstimator method) (torch_mist.estimators.transformed.TransformedMIEstimator method) (torch_mist.estimators.TransformedMIEstimator method) (torch_mist.estimators.utils.FlippedMIEstimator method) (torch_mist.models.bottleneck.InformationBottleneck method)

N

n_bins (torch_mist.estimators.hybrid.QuantizationFunction property) (torch_mist.estimators.QuantizationFunction property), [1] (torch_mist.estimators.transformed.QuantizationFunction property) (torch_mist.quantization.ClusterQuantization property) (torch_mist.quantization.FixedQuantization property) (torch_mist.quantization.functions.ClusterQuantization property) (torch_mist.quantization.functions.FixedQuantization property) (torch_mist.quantization.functions.QuantizationFunction property) (torch_mist.quantization.functions.VectorQuantization property) (torch_mist.quantization.QuantizationFunction property), [1] (torch_mist.quantization.VectorQuantization property) n_negatives_to_use() (torch_mist.estimators.discriminative.base.DiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.DiscriminativeMIEstimator method) (torch_mist.estimators.DiscriminativeMIEstimator method) (torch_mist.estimators.hybrid.DiscriminativeMIEstimator method), [1]

n_parameters (torch_mist.distributions.parametrizations.map.DeltaMap attribute) (torch_mist.distributions.parametrizations.map.LogitsMap attribute) (torch_mist.distributions.parametrizations.map.ParameterMap attribute) n_trainable_parameters() (in module torch_mist.utils) (in module torch_mist.utils.freeze) Normalize (class in torch_mist.nn) NormalModule (class in torch_mist.distributions) (class in torch_mist.distributions.normal) NotTrainedError NWJ (class in torch_mist.estimators), [1] (class in torch_mist.estimators.discriminative), [1] (class in torch_mist.estimators.discriminative.implementations) (class in torch_mist.estimators.discriminative.implementations.nwj) nwj() (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) (in module torch_mist.estimators.discriminative.factories)

O

on_epoch_end() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) on_epoch_start() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) on_iteration_end() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method)

on_iteration_start() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) on_split_end() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) on_split_start() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) ONE_HEAD (in module torch_mist.critic.factories)

P

PandasLogger (class in torch_mist.utils.logging) (class in torch_mist.utils.logging.logger) (class in torch_mist.utils.logging.logger.pandas) ParameterMap (class in torch_mist.distributions.parametrizations.map) parse_data() (in module torch_mist.utils.data.utils) Permute (class in torch_mist.distributions.transforms), [1] (class in torch_mist.distributions.transforms.implementations) (class in torch_mist.distributions.transforms.implementations.permute) permute() (in module torch_mist.distributions.transforms) (in module torch_mist.distributions.transforms.factories) POSSIBLE_HEADS (in module torch_mist.critic.factories) PQ (class in torch_mist.estimators), [1] (class in torch_mist.estimators.transformed), [1] (class in torch_mist.estimators.transformed.implementations) (class in torch_mist.estimators.transformed.implementations.pq)

pq() (in module torch_mist.estimators) (in module torch_mist.estimators.transformed) (in module torch_mist.estimators.transformed.factories) PQHybridMIEstimator (class in torch_mist.estimators) (class in torch_mist.estimators.hybrid) (class in torch_mist.estimators.hybrid.implementations) (class in torch_mist.estimators.hybrid.implementations.hybrid_pq) prepare_samples() (in module torch_mist.data.utils) prepare_variables() (in module torch_mist.utils.data.utils) pretrain_components() (in module torch_mist.utils.train.mi_estimator)

Q

q_X (torch_mist.estimators.generative.base.JointGenerativeMIEstimator property) (torch_mist.estimators.generative.GM property), [1] (torch_mist.estimators.generative.implementations.GM property) (torch_mist.estimators.generative.implementations.gm.GM property) (torch_mist.estimators.GM property), [1] q_Y (torch_mist.estimators.generative.base.JointGenerativeMIEstimator property) (torch_mist.estimators.generative.GM property), [1] (torch_mist.estimators.generative.implementations.GM property) (torch_mist.estimators.generative.implementations.gm.GM property) (torch_mist.estimators.GM property), [1] QuantizationFunction (class in torch_mist.estimators), [1] (class in torch_mist.estimators.hybrid) (class in torch_mist.estimators.transformed) (class in torch_mist.quantization), [1] (class in torch_mist.quantization.functions) quantize() (torch_mist.estimators.hybrid.QuantizationFunction method) (torch_mist.estimators.QuantizationFunction method), [1] (torch_mist.estimators.transformed.QuantizationFunction method) (torch_mist.models.vqvae.VQVAE method) (torch_mist.quantization.ClusterQuantization method) (torch_mist.quantization.FixedQuantization method) (torch_mist.quantization.functions.ClusterQuantization method) (torch_mist.quantization.functions.FixedQuantization method) (torch_mist.quantization.functions.LearnableVectorQuantization method) (torch_mist.quantization.functions.QuantizationFunction method) (torch_mist.quantization.functions.VectorQuantization method) (torch_mist.quantization.LearnableVectorQuantization method) (torch_mist.quantization.QuantizationFunction method), [1] (torch_mist.quantization.VectorQuantization method) (torch_mist.quantization.VQVAE method)

quantize_y (torch_mist.estimators.hybrid.implementations.hybrid_pq.PQHybridMIEstimator property) (torch_mist.estimators.hybrid.implementations.PQHybridMIEstimator property) (torch_mist.estimators.hybrid.PQHybridMIEstimator property) (torch_mist.estimators.PQHybridMIEstimator property)

R

regularization() (torch_mist.models.bottleneck.CEB method) (torch_mist.models.bottleneck.InformationBottleneck method) (torch_mist.models.bottleneck.MIB method) (torch_mist.models.bottleneck.VIB method) (torch_mist.models.CEB method) (torch_mist.models.MIB method) (torch_mist.models.VIB method) resampled_hybrid() (in module torch_mist.estimators) (in module torch_mist.estimators.hybrid) (in module torch_mist.estimators.hybrid.factories) ResampledHybridMIEstimator (class in torch_mist.estimators) (class in torch_mist.estimators.hybrid) (class in torch_mist.estimators.hybrid.implementations) (class in torch_mist.estimators.hybrid.implementations.resampled) resampling_strategy() (torch_mist.estimators.hybrid.base.HybridMIEstimator method) (torch_mist.estimators.hybrid.HybridMIEstimator method) (torch_mist.estimators.HybridMIEstimator method)

reset_log() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) reweighed_hybrid() (in module torch_mist.estimators) (in module torch_mist.estimators.hybrid) (in module torch_mist.estimators.hybrid.factories) ReweighedHybridMIEstimator (class in torch_mist.estimators) (class in torch_mist.estimators.hybrid) (class in torch_mist.estimators.hybrid.implementations) (class in torch_mist.estimators.hybrid.implementations.reweighted) rsample() (torch_mist.distributions.categorical.CategoricalModule method) (torch_mist.distributions.CategoricalModule method) (torch_mist.distributions.joint.wrapper.TorchJointDistribution method) (torch_mist.distributions.normal.NormalModule method) (torch_mist.distributions.NormalModule method) (torch_mist.distributions.transforms.base.TransformedDistributionModule method) (torch_mist.distributions.transforms.TransformedDistributionModule method) RunTerminationManager (class in torch_mist.utils.train.utils)

S

SameAttributeDataLoader (class in torch_mist.utils.data) (class in torch_mist.utils.data.loader) SameAttributeSampler (class in torch_mist.utils.data) (class in torch_mist.utils.data.sampler) sample() (torch_mist.distributions.empirical.EmpiricalDistribution method) (torch_mist.distributions.EmpiricalDistribution method) (torch_mist.distributions.joint.categorical.JointCategorical method) (torch_mist.distributions.joint.wrapper.TorchJointDistribution method) sample_negatives() (torch_mist.estimators.discriminative.base.DiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.DiscriminativeMIEstimator method) (torch_mist.estimators.DiscriminativeMIEstimator method) (torch_mist.estimators.hybrid.base.HybridMIEstimator method) (torch_mist.estimators.hybrid.DiscriminativeMIEstimator method), [1] (torch_mist.estimators.hybrid.HybridMIEstimator method) (torch_mist.estimators.hybrid.implementations.hybrid_pq.PQHybridMIEstimator method) (torch_mist.estimators.hybrid.implementations.PQHybridMIEstimator method) (torch_mist.estimators.hybrid.implementations.resampled.ResampledHybridMIEstimator method) (torch_mist.estimators.hybrid.implementations.ResampledHybridMIEstimator method) (torch_mist.estimators.hybrid.implementations.ReweighedHybridMIEstimator method) (torch_mist.estimators.hybrid.implementations.reweighted.ReweighedHybridMIEstimator method) (torch_mist.estimators.hybrid.PQHybridMIEstimator method) (torch_mist.estimators.hybrid.ResampledHybridMIEstimator method) (torch_mist.estimators.hybrid.ReweighedHybridMIEstimator method) (torch_mist.estimators.HybridMIEstimator method) (torch_mist.estimators.PQHybridMIEstimator method) (torch_mist.estimators.ResampledHybridMIEstimator method) (torch_mist.estimators.ReweighedHybridMIEstimator method) sample_same_attributes() (in module torch_mist.utils.data) (in module torch_mist.utils.data.loader) sample_same_value() (in module torch_mist.utils.data.loader) SampleDataset (class in torch_mist.utils.data) (class in torch_mist.utils.data.dataset) save_log() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method)

save_model() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) (torch_mist.utils.logging.logger.wandb.WandbLogger method) save_weights() (torch_mist.utils.train.utils.RunTerminationManager method) select_k_others() (in module torch_mist.utils.indexing) SEPARABLE_CRITIC (in module torch_mist.critic.base) (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) separable_critic() (in module torch_mist.critic) (in module torch_mist.critic.factories) SeparableCritic (class in torch_mist.critic) (class in torch_mist.critic.separable) shared_critic_nns() (in module torch_mist.critic.factories) (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) shared_joint_critics() (in module torch_mist.critic.factories) shared_separable_critics() (in module torch_mist.critic.factories) should_stop() (torch_mist.utils.train.utils.RunTerminationManager method) SMILE (class in torch_mist.estimators), [1] (class in torch_mist.estimators.discriminative), [1] (class in torch_mist.estimators.discriminative.implementations) (class in torch_mist.estimators.discriminative.implementations.smile) smile() (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) (in module torch_mist.estimators.discriminative.factories) SPLIT_SEQUENCE (in module torch_mist.estimators.transformed.base) StandardNormalModule (class in torch_mist.distributions) (class in torch_mist.distributions.normal) StochasticDimensionalityReduction (class in torch_mist.decomposition.base) supported_distributions (torch_mist.distributions.parametrizations.map.DeltaMap attribute) (torch_mist.distributions.parametrizations.map.LocScaleMap attribute) (torch_mist.distributions.parametrizations.map.LogitsMap attribute) (torch_mist.distributions.parametrizations.map.ParameterMap attribute) SYMMETRIC_HEADS (in module torch_mist.critic.factories)

T

T (in module torch_mist.distributions.joint.base) (in module torch_mist.utils.caching) temporal_offset_data() (in module torch_mist.utils.data.temporal) TemporalMIEstimator (class in torch_mist.estimators.temporal) TensorDictLike (in module torch_mist.utils.data.utils) test() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) TIB (class in torch_mist.decomposition) (class in torch_mist.decomposition.tib) (class in torch_mist.models.bottleneck) TInfoMax (class in torch_mist.decomposition.tinfomax) torch_mist module torch_mist.baseline module torch_mist.baseline.base module torch_mist.baseline.factories module torch_mist.critic module torch_mist.critic.base module torch_mist.critic.constant module torch_mist.critic.factories module torch_mist.critic.joint module torch_mist.critic.separable module torch_mist.data module torch_mist.data.multimixture module torch_mist.data.multivariate module torch_mist.data.utils module torch_mist.decomposition module torch_mist.decomposition.base module torch_mist.decomposition.ceb module torch_mist.decomposition.mib module torch_mist.decomposition.mid module torch_mist.decomposition.tib module torch_mist.decomposition.tinfomax module torch_mist.decomposition.vib module torch_mist.distributions module torch_mist.distributions.caching module torch_mist.distributions.categorical module torch_mist.distributions.conditional module torch_mist.distributions.conditional.delta module torch_mist.distributions.conditional.unconditional module torch_mist.distributions.empirical module torch_mist.distributions.factories module torch_mist.distributions.joint module torch_mist.distributions.joint.base module torch_mist.distributions.joint.categorical module torch_mist.distributions.joint.wrapper module torch_mist.distributions.normal module torch_mist.distributions.parametrizations module torch_mist.distributions.parametrizations.map module torch_mist.distributions.transforms module torch_mist.distributions.transforms.base module torch_mist.distributions.transforms.factories module torch_mist.distributions.transforms.implementations module torch_mist.distributions.transforms.implementations.linear module torch_mist.distributions.transforms.implementations.normalize module torch_mist.distributions.transforms.implementations.permute module torch_mist.distributions.transforms.utils module torch_mist.estimators module torch_mist.estimators.base module torch_mist.estimators.discriminative module torch_mist.estimators.discriminative.base module torch_mist.estimators.discriminative.factories module torch_mist.estimators.discriminative.implementations module torch_mist.estimators.discriminative.implementations.alpha_tuba module torch_mist.estimators.discriminative.implementations.dummy module torch_mist.estimators.discriminative.implementations.flo module torch_mist.estimators.discriminative.implementations.infonce module torch_mist.estimators.discriminative.implementations.js module torch_mist.estimators.discriminative.implementations.mine module torch_mist.estimators.discriminative.implementations.nwj module torch_mist.estimators.discriminative.implementations.smile module torch_mist.estimators.discriminative.implementations.tuba module torch_mist.estimators.factories module torch_mist.estimators.generative module torch_mist.estimators.generative.base module torch_mist.estimators.generative.factories module torch_mist.estimators.generative.implementations module torch_mist.estimators.generative.implementations.ba module torch_mist.estimators.generative.implementations.club module

torch_mist.estimators.generative.implementations.doe module torch_mist.estimators.generative.implementations.dummy module torch_mist.estimators.generative.implementations.gm module torch_mist.estimators.generative.implementations.l1out module torch_mist.estimators.hybrid module torch_mist.estimators.hybrid.base module torch_mist.estimators.hybrid.factories module torch_mist.estimators.hybrid.implementations module torch_mist.estimators.hybrid.implementations.hybrid_pq module torch_mist.estimators.hybrid.implementations.resampled module torch_mist.estimators.hybrid.implementations.reweighted module torch_mist.estimators.multi module torch_mist.estimators.temporal module torch_mist.estimators.transformed module torch_mist.estimators.transformed.base module torch_mist.estimators.transformed.factories module torch_mist.estimators.transformed.implementations module torch_mist.estimators.transformed.implementations.binned module torch_mist.estimators.transformed.implementations.flip module torch_mist.estimators.transformed.implementations.pq module torch_mist.estimators.transformed.utils module torch_mist.estimators.utils module torch_mist.models module torch_mist.models.bottleneck module torch_mist.models.vqvae module torch_mist.nn module torch_mist.quantization module torch_mist.quantization.factories module torch_mist.quantization.functions module torch_mist.utils module torch_mist.utils.caching module torch_mist.utils.data module torch_mist.utils.data.csv module torch_mist.utils.data.dataset module torch_mist.utils.data.loader module torch_mist.utils.data.sampler module torch_mist.utils.data.temporal module torch_mist.utils.data.utils module torch_mist.utils.estimation module torch_mist.utils.evaluation module torch_mist.utils.evaluation.mi_estimator module torch_mist.utils.evaluation.model module torch_mist.utils.freeze module torch_mist.utils.indexing module torch_mist.utils.logging module torch_mist.utils.logging.logger module torch_mist.utils.logging.logger.base module torch_mist.utils.logging.logger.pandas module torch_mist.utils.logging.logger.wandb module torch_mist.utils.logging.metrics module torch_mist.utils.shape module torch_mist.utils.train module torch_mist.utils.train.mi_estimator module torch_mist.utils.train.model module torch_mist.utils.train.utils module TorchJointDistribution (class in torch_mist.distributions.joint.wrapper) train() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) train_baseline() (torch_mist.estimators.discriminative.implementations.MINE method) (torch_mist.estimators.discriminative.implementations.mine.MINE method) (torch_mist.estimators.discriminative.MINE method), [1] (torch_mist.estimators.MINE method), [1] train_epoch() (in module torch_mist.utils.train.model) train_mi_estimator() (in module torch_mist.utils) (in module torch_mist.utils.train) (in module torch_mist.utils.train.mi_estimator) train_model() (in module torch_mist.utils.train) (in module torch_mist.utils.train.model) transform() (torch_mist.decomposition.base.DimensionalityReduction method) (torch_mist.estimators.transformed.base.TransformedMIEstimator method) (torch_mist.estimators.transformed.TransformedMIEstimator method) (torch_mist.estimators.TransformedMIEstimator method) transformed_normal() (in module torch_mist.distributions) (in module torch_mist.distributions.factories) TransformedDistributionModule (class in torch_mist.distributions.transforms) (class in torch_mist.distributions.transforms.base) TransformedMIEstimator (class in torch_mist.estimators) (class in torch_mist.estimators.transformed) (class in torch_mist.estimators.transformed.base) TransformedNormalModule (class in torch_mist.distributions), [1] (class in torch_mist.distributions.normal) transition (torch_mist.decomposition.TIB property) (torch_mist.decomposition.tib.TIB property) TUBA (class in torch_mist.estimators), [1] (class in torch_mist.estimators.discriminative), [1] (class in torch_mist.estimators.discriminative.implementations) (class in torch_mist.estimators.discriminative.implementations.tuba) tuba() (in module torch_mist.estimators) (in module torch_mist.estimators.discriminative) (in module torch_mist.estimators.discriminative.factories)

U

UnconditionalDistribution (class in torch_mist.distributions.conditional.unconditional) unnormalized_log_ratio() (torch_mist.estimators.base.MIEstimator method) (torch_mist.estimators.discriminative.base.DiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.DiscriminativeMIEstimator method) (torch_mist.estimators.discriminative.FLO method), [1] (torch_mist.estimators.discriminative.implementations.FLO method) (torch_mist.estimators.discriminative.implementations.flo.FLO method) (torch_mist.estimators.DiscriminativeMIEstimator method) (torch_mist.estimators.FLO method), [1] (torch_mist.estimators.generative.base.GenerativeMIEstimator method) (torch_mist.estimators.generative.GenerativeMIEstimator method) (torch_mist.estimators.GenerativeMIEstimator method) (torch_mist.estimators.hybrid.base.HybridMIEstimator method) (torch_mist.estimators.hybrid.DiscriminativeMIEstimator method), [1] (torch_mist.estimators.hybrid.GenerativeMIEstimator method) (torch_mist.estimators.hybrid.HybridMIEstimator method) (torch_mist.estimators.hybrid.MIEstimator method), [1] (torch_mist.estimators.HybridMIEstimator method) (torch_mist.estimators.MIEstimator method), [1], [2] (torch_mist.estimators.transformed.base.TransformedMIEstimator method) (torch_mist.estimators.transformed.implementations.flip.FlippedMIEstimator method) (torch_mist.estimators.transformed.TransformedMIEstimator method) (torch_mist.estimators.TransformedMIEstimator method)

update() (torch_mist.distributions.empirical.EmpiricalDistribution method) (torch_mist.distributions.EmpiricalDistribution method) update_dataloader() (in module torch_mist.utils.data.utils) update_device() (torch_mist.distributions.transforms.implementations.Permute method) (torch_mist.distributions.transforms.implementations.permute.Permute method) (torch_mist.distributions.transforms.Permute method), [1] upper_bound (torch_mist.models.vqvae.VQVAE attribute) (torch_mist.nn.Model attribute) (torch_mist.quantization.VQVAE attribute)

V

valid() (torch_mist.utils.logging.Logger method) (torch_mist.utils.logging.logger.base.Logger method) (torch_mist.utils.logging.logger.Logger method) validate() (in module torch_mist.utils.train.model) VectorQuantization (class in torch_mist.quantization) (class in torch_mist.quantization.functions) VERSIONS (in module torch_mist.models.vqvae)

VIB (class in torch_mist.decomposition) (class in torch_mist.decomposition.vib) (class in torch_mist.models) (class in torch_mist.models.bottleneck) VQVAE (class in torch_mist.models.vqvae) (class in torch_mist.quantization) vqvae() (in module torch_mist.quantization) (in module torch_mist.quantization.factories)

W

WandbLogger (class in torch_mist.utils.logging.logger.wandb)

WrappedDataset (class in torch_mist.utils.data.dataset)