jax.debug_infs#
- jax.debug_infs = <jax._src.config.State object>#
Context manager for jax_debug_infs config option.
Add inf checks to every operation. When an inf is detected on the output of a jit-compiled computation, call into the un-compiled version in an attempt to more precisely identify the operation which produced the inf.
- Parameters:
new_val (Any)