torch_blue
Home
Installation
Quickstart
Debug Guide
API Reference
torch_blue
torch_blue.vi
torch_blue.vi.distributions
torch_blue.vi.utils
torch_blue.vi.utils.NoVariablesError
torch_blue.vi.utils.UnsupportedDistributionError
torch_blue.vi.utils.PostInitCallMeta
torch_blue.vi.AnalyticalKullbackLeiblerLoss
torch_blue.vi.KullbackLeiblerModule
torch_blue.vi.NonBayesianDivergence
torch_blue.vi.NormalNormalDivergence
torch_blue.vi.UniformNormalDivergence
torch_blue.vi.VIModule
torch_blue.vi.VIConv1d
torch_blue.vi.VIConv2d
torch_blue.vi.VIConv3d
torch_blue.vi.KullbackLeiblerLoss
torch_blue.vi.VILinear
torch_blue.vi.VIResidualConnection
torch_blue.vi.VISequential
torch_blue.vi.VIMultiheadAttention
torch_blue.vi.VITransformer
torch_blue.vi.VITransformerDecoder
torch_blue.vi.VITransformerDecoderLayer
torch_blue.vi.VITransformerEncoder
torch_blue.vi.VITransformerEncoderLayer
torch_blue.vi.VIkwargs
torch_blue.vi.VIReturn
torch_blue
API Reference
torch_blue
torch_blue.vi
torch_blue.vi.utils
torch_blue.vi.utils.PostInitCallMeta
View page source
torch_blue.vi.utils.PostInitCallMeta
class
torch_blue.vi.utils.
PostInitCallMeta
Bases:
type
Metaclass calling __post_init__ after __init__.