scvi.external.mrvi\_jax.JaxMRVAE
================================

.. currentmodule:: scvi.external.mrvi_jax

.. add toctree option to make autodoc generate the pages

.. autoclass:: JaxMRVAE
   :show-inheritance:



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

.. autosummary::

        ~scvi.external.mrvi_jax.JaxMRVAE.encoder_n_hidden
        ~scvi.external.mrvi_jax.JaxMRVAE.encoder_n_layers
        ~scvi.external.mrvi_jax.JaxMRVAE.learn_z_u_prior_scale
        ~scvi.external.mrvi_jax.JaxMRVAE.n_latent
        ~scvi.external.mrvi_jax.JaxMRVAE.n_latent_u
        ~scvi.external.mrvi_jax.JaxMRVAE.n_obs_per_sample
        ~scvi.external.mrvi_jax.JaxMRVAE.name
        ~scvi.external.mrvi_jax.JaxMRVAE.parent
        ~scvi.external.mrvi_jax.JaxMRVAE.px_kwargs
        ~scvi.external.mrvi_jax.JaxMRVAE.qu_kwargs
        ~scvi.external.mrvi_jax.JaxMRVAE.qz_kwargs
        ~scvi.external.mrvi_jax.JaxMRVAE.required_rngs
        ~scvi.external.mrvi_jax.JaxMRVAE.scale_observations
        ~scvi.external.mrvi_jax.JaxMRVAE.scope
        ~scvi.external.mrvi_jax.JaxMRVAE.training
        ~scvi.external.mrvi_jax.JaxMRVAE.u_prior_mixture
        ~scvi.external.mrvi_jax.JaxMRVAE.u_prior_mixture_k
        ~scvi.external.mrvi_jax.JaxMRVAE.u_prior_scale
        ~scvi.external.mrvi_jax.JaxMRVAE.z_u_prior
        ~scvi.external.mrvi_jax.JaxMRVAE.z_u_prior_scale
        ~scvi.external.mrvi_jax.JaxMRVAE.n_input
        ~scvi.external.mrvi_jax.JaxMRVAE.n_sample
        ~scvi.external.mrvi_jax.JaxMRVAE.n_batch
        ~scvi.external.mrvi_jax.JaxMRVAE.n_labels






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

.. autosummary::

    ~scvi.external.mrvi_jax.JaxMRVAE.compute_h_from_x_eps
    ~scvi.external.mrvi_jax.JaxMRVAE.generative
    ~scvi.external.mrvi_jax.JaxMRVAE.inference
    ~scvi.external.mrvi_jax.JaxMRVAE.loss
    ~scvi.external.mrvi_jax.JaxMRVAE.setup





Attributes
~~~~~~~~~~



.. autoattribute:: JaxMRVAE.encoder_n_hidden

.. autoattribute:: JaxMRVAE.encoder_n_layers

.. autoattribute:: JaxMRVAE.learn_z_u_prior_scale

.. autoattribute:: JaxMRVAE.n_latent

.. autoattribute:: JaxMRVAE.n_latent_u

.. autoattribute:: JaxMRVAE.n_obs_per_sample

.. autoattribute:: JaxMRVAE.name

.. autoattribute:: JaxMRVAE.parent

.. autoattribute:: JaxMRVAE.px_kwargs

.. autoattribute:: JaxMRVAE.qu_kwargs

.. autoattribute:: JaxMRVAE.qz_kwargs

.. autoattribute:: JaxMRVAE.required_rngs

.. autoattribute:: JaxMRVAE.scale_observations

.. autoattribute:: JaxMRVAE.scope

.. autoattribute:: JaxMRVAE.training

.. autoattribute:: JaxMRVAE.u_prior_mixture

.. autoattribute:: JaxMRVAE.u_prior_mixture_k

.. autoattribute:: JaxMRVAE.u_prior_scale

.. autoattribute:: JaxMRVAE.z_u_prior

.. autoattribute:: JaxMRVAE.z_u_prior_scale

.. autoattribute:: JaxMRVAE.n_input

.. autoattribute:: JaxMRVAE.n_sample

.. autoattribute:: JaxMRVAE.n_batch

.. autoattribute:: JaxMRVAE.n_labels






Methods
~~~~~~~



.. automethod:: JaxMRVAE.compute_h_from_x_eps

.. automethod:: JaxMRVAE.generative

.. automethod:: JaxMRVAE.inference

.. automethod:: JaxMRVAE.loss

.. automethod:: JaxMRVAE.setup