Content - 198086a8fa7aa548936b05c8faf489ec9ad0b3cd - e7d6bb3/docs/jax.config.rst

visit type:

Tip revision: 0118f8d568d4d94665e01f284f63f0a5d9865f5e authored by Yash Katariya on 08 December 2022, 00:01:43 UTC
Prepare for jax and jaxlib 0.4.0 release

Tip revision: 0118f8d

jax.config.rst

.. currentmodule:: jax

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

.. autosummary::
   :toctree: _autosummary

   config
   check_tracer_leaks
   checking_leaks
   debug_nans
   debug_infs
   default_device
   default_matmul_precision
   default_prng_impl
   enable_checks
   enable_custom_prng
   enable_custom_vjp_by_custom_transpose
   log_compiles
   numpy_rank_promotion
   transfer_guard

Browse the archive

https://github.com/google/jax