jax.debug module# Runtime value debugging utilities#
Compiled prints and breakpoints describes how to make use of JAX’s runtime value debugging features.
callback(callback, *args[, ordered, partitioned])
Calls a stageable Python callback.
print(fmt, *args[, ordered, partitioned])
Prints values and works in staged out JAX functions.
breakpoint(*[, backend, filter_frames, ...])
Enters a breakpoint at a point in a program.
Sharding debugging utilities#Functions that enable inspecting and visualizing array shardings inside (and outside) staged functions.
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4