is_tracing

t3toolbox.safety.is_tracing(*arrays)

def is_tracing(*arrays):

True if we are inside a jax transform (jit / grad / vmap): any argument is a tracer, or we are globally under a trace (the closed-over-concrete-operand case – see _inside_jax_trace()).