is_tracing

t3toolbox.safety.is_tracing(*arrays)

def is_tracing(*arrays):

True if we are inside a jax transform: any argument is a tracer (catches jit / grad / vmap operands), or we are inside a jit trace with only closed-over concrete operands (the probe – see _inside_jax_trace()).