jax.debug_infs

Contents

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)