def broken_debug(x):
    print(f"x = {x}")       # Prints an abstract tracer, not a value
    if x > 0:                # Can crash, tracers don't have a boolean truth value
        return x * 2
    return x

jax.jit(broken_debug)(5.0)

        __
        __