scvi.module.base.JaxBaseModuleClass
===================================

.. currentmodule:: scvi.module.base

.. add toctree option to make autodoc generate the pages

.. autoclass:: JaxBaseModuleClass
   :show-inheritance:



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

.. autosummary::

        ~scvi.module.base.JaxBaseModuleClass.device
        ~scvi.module.base.JaxBaseModuleClass.name
        ~scvi.module.base.JaxBaseModuleClass.params
        ~scvi.module.base.JaxBaseModuleClass.parent
        ~scvi.module.base.JaxBaseModuleClass.required_rngs
        ~scvi.module.base.JaxBaseModuleClass.rngs
        ~scvi.module.base.JaxBaseModuleClass.scope
        ~scvi.module.base.JaxBaseModuleClass.state






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

.. autosummary::

    ~scvi.module.base.JaxBaseModuleClass.as_bound
    ~scvi.module.base.JaxBaseModuleClass.as_numpy_array
    ~scvi.module.base.JaxBaseModuleClass.configure
    ~scvi.module.base.JaxBaseModuleClass.eval
    ~scvi.module.base.JaxBaseModuleClass.generative
    ~scvi.module.base.JaxBaseModuleClass.get_jit_inference_fn
    ~scvi.module.base.JaxBaseModuleClass.inference
    ~scvi.module.base.JaxBaseModuleClass.load_state_dict
    ~scvi.module.base.JaxBaseModuleClass.loss
    ~scvi.module.base.JaxBaseModuleClass.on_load
    ~scvi.module.base.JaxBaseModuleClass.setup
    ~scvi.module.base.JaxBaseModuleClass.state_dict
    ~scvi.module.base.JaxBaseModuleClass.to
    ~scvi.module.base.JaxBaseModuleClass.train





Attributes
~~~~~~~~~~



.. autoattribute:: JaxBaseModuleClass.device

.. autoattribute:: JaxBaseModuleClass.name

.. autoattribute:: JaxBaseModuleClass.params

.. autoattribute:: JaxBaseModuleClass.parent

.. autoattribute:: JaxBaseModuleClass.required_rngs

.. autoattribute:: JaxBaseModuleClass.rngs

.. autoattribute:: JaxBaseModuleClass.scope

.. autoattribute:: JaxBaseModuleClass.state






Methods
~~~~~~~



.. automethod:: JaxBaseModuleClass.as_bound

.. automethod:: JaxBaseModuleClass.as_numpy_array

.. automethod:: JaxBaseModuleClass.configure

.. automethod:: JaxBaseModuleClass.eval

.. automethod:: JaxBaseModuleClass.generative

.. automethod:: JaxBaseModuleClass.get_jit_inference_fn

.. automethod:: JaxBaseModuleClass.inference

.. automethod:: JaxBaseModuleClass.load_state_dict

.. automethod:: JaxBaseModuleClass.loss

.. automethod:: JaxBaseModuleClass.on_load

.. automethod:: JaxBaseModuleClass.setup

.. automethod:: JaxBaseModuleClass.state_dict

.. automethod:: JaxBaseModuleClass.to

.. automethod:: JaxBaseModuleClass.train