.. _notebook_gallery:

Notebook gallery
================


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


.. raw:: html

    <div class="sphx-glr-thumbnails">
    
    <div class="sphx-glr-thumbcontainer" tooltip="Resnet example with Flax and JAXopt.">

.. only:: html

 .. figure:: /notebooks/deep_learning/thumbnails/resnet_flax.png
     :alt: Resnet example with Flax and JAXopt.

     :doc:`/notebooks/deep_learning/resnet_flax`

.. raw:: html

    </div>



.. raw:: html

    <div class="sphx-glr-thumbcontainer" tooltip="Adversarial training with JAXopt.">

.. only:: html

 .. figure:: /notebooks/deep_learning/thumbnails/adversarial_training.png
     :alt: Adversarial Training.

     :doc:`/notebooks/deep_learning/adversarial_training`

.. raw:: html

    </div>

    </div><!-- end sphx-glr-thumbnails -->




Implicit Differentiation
------------------------


.. raw:: html

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

    <div class="sphx-glr-thumbcontainer" tooltip="Bi-level dataset distillation with JAXopt.">

.. only:: html

 .. figure:: /notebooks/implicit_diff/thumbnails/plot_dataset_distillation.png
     :alt: Dataset distillation example with JAXopt.

     :doc:`/notebooks/implicit_diff/dataset_distillation`

.. raw:: html

    </div>


.. raw:: html

    <div class="sphx-glr-thumbcontainer" tooltip="Few-shot Adaptation with Model Agnostic Meta-Learning (MAML)">

.. only:: html

 .. figure:: /notebooks/implicit_diff/thumbnails/maml.png
     :alt: Few-shot Adaptation with Model Agnostic Meta-Learning (MAML)

     :doc:`/notebooks/implicit_diff/maml`

.. raw:: html

    </div>

    </div><!-- end sphx-glr-thumbnails -->



Distributed Optimization
------------------------


.. raw:: html

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

    <div class="sphx-glr-thumbcontainer" tooltip="Distributed Optimization with JAXopt.">

.. only:: html

 .. figure:: /notebooks/distributed/thumbnails/plot_custom_loop_pjit_example.png
     :alt: `jax.experimental.pjit` example using JAXopt.

     :doc:`/notebooks/distributed/custom_loop_pjit_example`

.. raw:: html

    </div>


.. raw:: html

    <div class="sphx-glr-thumbcontainer" tooltip="pmap example using jaxopt">

.. only:: html

 .. figure:: /notebooks/distributed/thumbnails/plot_custom_loop_pmap_example.png
     :alt: `jax.pmap` example using JAXopt.

     :doc:`/notebooks/distributed/custom_loop_pmap_example`


.. raw:: html

    </div>

    </div><!-- end sphx-glr-thumbnails -->


Perturbed optimizers
--------------------


.. raw:: html

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

    <div class="sphx-glr-thumbcontainer" tooltip="Perturbed optimizers with JAXopt.">

.. only:: html

 .. figure:: /notebooks/perturbed_optimizers/thumbnails/perturbations.png
     :alt: Perturbed optimizers with JAXopt.

     :doc:`/notebooks/perturbed_optimizers/perturbed_optimizers`

.. raw:: html

    </div>

    </div><!-- end sphx-glr-thumbnails -->