mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 11:56:07 +00:00
34 lines
630 B
ReStructuredText
34 lines
630 B
ReStructuredText
|
|
``jax.debug`` module
|
|
====================
|
|
|
|
.. currentmodule:: jax.debug
|
|
|
|
.. automodule:: jax.debug
|
|
|
|
Runtime value debugging utilities
|
|
---------------------------------
|
|
|
|
:doc:`debugging/print_breakpoint` describes how to make use of JAX's runtime value
|
|
debugging features.
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
callback
|
|
print
|
|
breakpoint
|
|
|
|
Sharding debugging utilities
|
|
----------------------------
|
|
|
|
Functions that enable inspecting and visualizing array shardings inside (and outside)
|
|
staged functions.
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
inspect_array_sharding
|
|
visualize_array_sharding
|
|
visualize_sharding
|