Remove compiler disable

Author: Ian Schweer

pytensor/link/pytorch/dispatch/scalar.py

@@ -55,7 +55,7 @@ def cast(x):
 
 @pytorch_funcify.register(ScalarLoop)
 def pytorch_funicify_ScalarLoop(op, node, **kwargs):
-    update = pytorch_funcify(op.fgraph)
+    update = pytorch_funcify(op.fgraph, **kwargs)
     state_length = op.nout
     if op.is_while:
 
@@ -84,4 +84,4 @@ def scalar_loop(steps, *start_and_constants):
             else:
                 return carry
 
-    return torch.compiler.disable(scalar_loop, recursive=False)
+    return scalar_loop