diff --git a/thunder/core/jit_ext.py b/thunder/core/jit_ext.py index ce774c4936..726353983a 100644 --- a/thunder/core/jit_ext.py +++ b/thunder/core/jit_ext.py @@ -778,7 +778,6 @@ def _general_jit_torch_ops_higher_order_autograd_function_apply(fwd, bwd, *fwd_a from thunder.core.baseutils import sequencify from thunder.core.pytree import tree_map from thunder.core.trace_interpreter import interpret_trace - from thunder.torch import autograd_function_apply def _generate_random_str_id() -> str: import secrets