jax.debug_infs¶

jax.debug_infs = <jax._src.config._StateContextManager 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.