diff --git a/src/torchjd/autogram/_module_hook_manager.py b/src/torchjd/autogram/_module_hook_manager.py
index e5087bb0..07381184 100644
--- a/src/torchjd/autogram/_module_hook_manager.py
+++ b/src/torchjd/autogram/_module_hook_manager.py
@@ -108,6 +108,8 @@ class JacobianAccumulator(torch.autograd.Function):
             toggle mechanism to activate only during the Gramian accumulation phase.
             """
 
+            generate_vmap_rule = True
+
             @staticmethod
             def forward(*xs: Tensor) -> tuple[Tensor, ...]:
                 return tuple([x.detach() for x in xs])