Skip to content

Latest commit

 

History

History
36 lines (26 loc) · 550 Bytes

user_guides.rst

File metadata and controls

36 lines (26 loc) · 550 Bytes

User Guides

User guides are deeper dives into particular topics within JAX that become relevant as your JAX project matures into larger or deployed codebases.

.. toctree::
   :maxdepth: 1
   :caption: Debugging and Performance

   profiling
   device_memory_profiling
   debugging/index


.. toctree::
   :maxdepth: 1
   :caption: Development

   jaxpr
   notebooks/external_callbacks
   type_promotion
   pytrees

.. toctree::
   :maxdepth: 1
   :caption: Run Time

   aot
   errors
   transfer_guard

   pallas/index