mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 12:56:07 +00:00
22 lines
370 B
ReStructuredText
22 lines
370 B
ReStructuredText
.. currentmodule:: jax
|
|
|
|
JAX configuration
|
|
=================
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
config
|
|
check_tracer_leaks
|
|
checking_leaks
|
|
debug_nans
|
|
debug_infs
|
|
default_matmul_precision
|
|
default_prng_impl
|
|
enable_checks
|
|
enable_custom_prng
|
|
enable_custom_vjp_by_custom_transpose
|
|
log_compiles
|
|
numpy_rank_promotion
|
|
transfer_guard
|