mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 12:56:07 +00:00
f26b866e08
This currently only supports setting a specific Device object, not a platform like "cpu". That should be added in the future. Bumps the minimum jaxlib version in order to include https://github.com/tensorflow/tensorflow/pull/53656
23 lines
388 B
ReStructuredText
23 lines
388 B
ReStructuredText
.. 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
|