.. _sphx_glr_auto_examples_deep_learning:

.. _deep_learning_examples:

Deep learning
-------------




.. raw:: html

    <div class="sphx-glr-thumbnails">


.. raw:: html

    <div class="sphx-glr-thumbcontainer" tooltip="VAE example with Haiku and JAXopt.">

.. only:: html

  .. image:: /auto_examples/deep_learning/images/thumb/sphx_glr_haiku_vae_thumb.png
    :alt: VAE example with Haiku and JAXopt.

  :ref:`sphx_glr_auto_examples_deep_learning_haiku_vae.py`

.. raw:: html

      <div class="sphx-glr-thumbnail-title">VAE example with Haiku and JAXopt.</div>
    </div>


.. raw:: html

    <div class="sphx-glr-thumbcontainer" tooltip="Image classification example with Haiku and JAXopt.">

.. only:: html

  .. image:: /auto_examples/deep_learning/images/thumb/sphx_glr_haiku_image_classif_thumb.png
    :alt: Image classification example with Haiku and JAXopt.

  :ref:`sphx_glr_auto_examples_deep_learning_haiku_image_classif.py`

.. raw:: html

      <div class="sphx-glr-thumbnail-title">Image classification example with Haiku and JAXopt.</div>
    </div>


.. raw:: html

    <div class="sphx-glr-thumbcontainer" tooltip="The purpose of this example is to illustrate the power of adaptive stepsize algorithms.">

.. only:: html

  .. image:: /auto_examples/deep_learning/images/thumb/sphx_glr_plot_sgd_solvers_thumb.png
    :alt: Comparison of different SGD algorithms.

  :ref:`sphx_glr_auto_examples_deep_learning_plot_sgd_solvers.py`

.. raw:: html

      <div class="sphx-glr-thumbnail-title">Comparison of different SGD algorithms.</div>
    </div>


.. raw:: html

    <div class="sphx-glr-thumbcontainer" tooltip="The purpose of this example is to illustrate how JAXopt solvers can be easily used for distribu...">

.. only:: html

  .. image:: /auto_examples/deep_learning/images/thumb/sphx_glr_distributed_flax_imagenet_thumb.png
    :alt: SPMD ResNet example with Flax and JAXopt.

  :ref:`sphx_glr_auto_examples_deep_learning_distributed_flax_imagenet.py`

.. raw:: html

      <div class="sphx-glr-thumbnail-title">SPMD ResNet example with Flax and JAXopt.</div>
    </div>


.. raw:: html

    </div>


.. toctree::
   :hidden:

   /auto_examples/deep_learning/haiku_vae
   /auto_examples/deep_learning/haiku_image_classif
   /auto_examples/deep_learning/plot_sgd_solvers
   /auto_examples/deep_learning/distributed_flax_imagenet