drinx.is_traced#
- drinx.is_traced(x)[source]#
Checks if an object is a JAX Tracer.
In JAX, tracers are used during transformations (like jit, grad, or vmap) to represent abstract values rather than concrete arrays. This function identifies if the input is currently being tracked by the JAX dispatcher.
- Parameters:
x (
Any) – The object to check.- Return type:
bool- Returns:
True if x is a jax.core.Tracer, False otherwise.