scvi.module.JaxVAE
==================

.. currentmodule:: scvi.module

.. add toctree option to make autodoc generate the pages

.. autoclass:: JaxVAE
   :show-inheritance:



Attributes table
~~~~~~~~~~~~~~~~

.. autosummary::

        ~scvi.module.JaxVAE.dropout_rate
        ~scvi.module.JaxVAE.eps
        ~scvi.module.JaxVAE.gene_likelihood
        ~scvi.module.JaxVAE.n_hidden
        ~scvi.module.JaxVAE.n_latent
        ~scvi.module.JaxVAE.n_layers
        ~scvi.module.JaxVAE.name
        ~scvi.module.JaxVAE.parent
        ~scvi.module.JaxVAE.required_rngs
        ~scvi.module.JaxVAE.scope
        ~scvi.module.JaxVAE.training
        ~scvi.module.JaxVAE.n_input
        ~scvi.module.JaxVAE.n_batch






Methods table
~~~~~~~~~~~~~~

.. autosummary::

    ~scvi.module.JaxVAE.generative
    ~scvi.module.JaxVAE.inference
    ~scvi.module.JaxVAE.loss
    ~scvi.module.JaxVAE.setup





Attributes
~~~~~~~~~~



.. autoattribute:: JaxVAE.dropout_rate

.. autoattribute:: JaxVAE.eps

.. autoattribute:: JaxVAE.gene_likelihood

.. autoattribute:: JaxVAE.n_hidden

.. autoattribute:: JaxVAE.n_latent

.. autoattribute:: JaxVAE.n_layers

.. autoattribute:: JaxVAE.name

.. autoattribute:: JaxVAE.parent

.. autoattribute:: JaxVAE.required_rngs

.. autoattribute:: JaxVAE.scope

.. autoattribute:: JaxVAE.training

.. autoattribute:: JaxVAE.n_input

.. autoattribute:: JaxVAE.n_batch






Methods
~~~~~~~



.. automethod:: JaxVAE.generative

.. automethod:: JaxVAE.inference

.. automethod:: JaxVAE.loss

.. automethod:: JaxVAE.setup