[docstring] compile -> jit in example (#1409)

Lightning-AI · Nov 11, 2024 · db84e15 · db84e15
1 parent a0a6151
commit db84e15
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/thunder/core/transforms.py b/thunder/core/transforms.py
@@ -2969,12 +2969,12 @@ def forward_and_backward_from_trace(trace: Trace, torch_autograd=False) -> Forwa
 
     Example:
         >>> import torch
-        >>> from thunder import compile, last_traces
+        >>> from thunder import jit, last_traces
         >>> from thunder.core.transforms import forward_and_backward_from_trace
         >>> def f(x):
         ...     return torch.sin(x)
         >>> x = torch.tensor(3.0)
-        >>> cf = compile(f)
+        >>> cf = jit(f)
         >>> out = cf(x)
         >>> trace = last_traces(cf)[0]
         >>> forward_and_backward_from_trace(trace)