.. automodule:: pyraug.models.base
.. autoclass:: pyraug.models.base.base_config.BaseModelConfig :members:
.. autoclass:: pyraug.models.BaseVAE :members:
.. autoclass:: pyraug.models.base.base_config.BaseSamplerConfig :members:
.. autoclass:: pyraug.models.base.base_sampler.BaseSampler :members: