.. currentmodule:: jax

JAX configuration
=================

.. autosummary::
   :toctree: _autosummary

   config
   enable_checks
   check_tracer_leaks
   checking_leaks
   enable_custom_prng
   debug_nans
   debug_infs
   log_compiles
   default_matmul_precision
   default_prng_impl
   numpy_rank_promotion