scvi.train.JaxTrainingPlanConfig
================================

.. currentmodule:: scvi.train

.. add toctree option to make autodoc generate the pages

.. autoclass:: JaxTrainingPlanConfig



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

.. autosummary::

    ~scvi.train.JaxTrainingPlanConfig.eps
    ~scvi.train.JaxTrainingPlanConfig.lr
    ~scvi.train.JaxTrainingPlanConfig.max_norm
    ~scvi.train.JaxTrainingPlanConfig.n_epochs_kl_warmup
    ~scvi.train.JaxTrainingPlanConfig.n_steps_kl_warmup
    ~scvi.train.JaxTrainingPlanConfig.optimizer
    ~scvi.train.JaxTrainingPlanConfig.optimizer_creator
    ~scvi.train.JaxTrainingPlanConfig.weight_decay
    ~scvi.train.JaxTrainingPlanConfig.loss_kwargs





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

.. autosummary::

    ~scvi.train.JaxTrainingPlanConfig.to_kwargs





Attributes
~~~~~~~~~~~



.. autoattribute:: JaxTrainingPlanConfig.eps

.. autoattribute:: JaxTrainingPlanConfig.lr

.. autoattribute:: JaxTrainingPlanConfig.max_norm

.. autoattribute:: JaxTrainingPlanConfig.n_epochs_kl_warmup

.. autoattribute:: JaxTrainingPlanConfig.n_steps_kl_warmup

.. autoattribute:: JaxTrainingPlanConfig.optimizer

.. autoattribute:: JaxTrainingPlanConfig.optimizer_creator

.. autoattribute:: JaxTrainingPlanConfig.weight_decay

.. autoattribute:: JaxTrainingPlanConfig.loss_kwargs






Methods
~~~~~~~



.. automethod:: JaxTrainingPlanConfig.to_kwargs