Module
======

Neural network components for the iVAE architecture.

.. autosummary::
    :toctree: doc
    :nosignatures:

    iVAE.module.weight_init
    iVAE.module.Encoder
    iVAE.module.Decoder
    iVAE.module.VAE