###########################
 JAX Privacy documentation
###########################

.. include:: introduction.md
   :parser: myst_parser.sphinx_

.. toctree::
   :maxdepth: 2
   :caption: Getting Started

   overview
   installation

.. toctree::
   :maxdepth: 2
   :caption: API Documentation

   jax_api
   keras_api

.. toctree::
   :maxdepth: 2
   :caption: Examples

   flax_linen_example
   gemma_dp_finetuning_example

.. toctree::
   :maxdepth: 2
   :caption: Paper Results Reproduction

   paper_reproductions

.. toctree::
   :maxdepth: 2
   :caption: Technical Documentation

   troubleshooting
   library_design
   contribution_guide
   support