- jax.debug.breakpoint(*, backend=None, filter_frames=True, num_frames=None, ordered=False, **kwargs)#
Enters a breakpoint at a point in a program.
bool) – Whether or not to filter out JAX-internal stack frames from the traceback. Since some libraries, like Flax, also make user of JAX’s stack frame filtering system, this option can also affect whether stack frames from libraries are filtered.
bool) – A keyword only argument used to indicate whether or not the staged out computation will enforce ordering of this
debug_printwith respect to other ordered