jax.debug package
=================

.. currentmodule:: jax.debug

.. automodule:: jax.debug

Debugging utilities
--------------------------

:doc:`debugging/print_breakpoint` describes how to make use of JAX's debugging features.

.. autosummary::
  :toctree: _autosummary

  callback
  print
  breakpoint