API Reference

This page contains auto-generated API reference documentation [1].

• 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.conditional
      • torch_mist.distributions.conditional.delta
      • torch_mist.distributions.conditional.unconditional
    • torch_mist.distributions.joint
      • torch_mist.distributions.joint.base
      • torch_mist.distributions.joint.categorical
      • torch_mist.distributions.joint.wrapper
    • torch_mist.distributions.parametrizations
      • torch_mist.distributions.parametrizations.map
    • torch_mist.distributions.transforms
      • 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.base
      • torch_mist.distributions.transforms.factories
      • torch_mist.distributions.transforms.utils
    • torch_mist.distributions.caching
    • torch_mist.distributions.categorical
    • torch_mist.distributions.empirical
    • torch_mist.distributions.factories
    • torch_mist.distributions.normal
  • torch_mist.estimators
    • torch_mist.estimators.discriminative
      • 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.discriminative.base
      • torch_mist.estimators.discriminative.factories
    • torch_mist.estimators.generative
      • 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.generative.base
      • torch_mist.estimators.generative.factories
    • torch_mist.estimators.hybrid
      • 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.hybrid.base
      • torch_mist.estimators.hybrid.factories
    • torch_mist.estimators.transformed
      • 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.base
      • torch_mist.estimators.transformed.factories
      • torch_mist.estimators.transformed.utils
    • torch_mist.estimators.base
    • torch_mist.estimators.factories
    • torch_mist.estimators.multi
    • torch_mist.estimators.temporal
    • torch_mist.estimators.utils
  • torch_mist.models
    • torch_mist.models.bottleneck
    • torch_mist.models.vqvae
  • torch_mist.quantization
    • torch_mist.quantization.factories
    • torch_mist.quantization.functions
  • torch_mist.utils
    • 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.evaluation
      • torch_mist.utils.evaluation.mi_estimator
      • torch_mist.utils.evaluation.model
    • 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.train
      • torch_mist.utils.train.mi_estimator
      • torch_mist.utils.train.model
      • torch_mist.utils.train.utils
    • torch_mist.utils.caching
    • torch_mist.utils.estimation
    • torch_mist.utils.freeze
    • torch_mist.utils.indexing
    • torch_mist.utils.shape
  • torch_mist.nn

[1]

Created with sphinx-autoapi