We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d7bbfae commit e666c49Copy full SHA for e666c49
src/torchjd/autojac/_jac.py
@@ -67,7 +67,7 @@ def jac(
67
<https://github.com/pytorch/pytorch/issues/138422>`_, `when some tensors have
68
<https://github.com/TorchJD/torchjd/issues/184>`_ ``retains_grad=True`` or `when using an
69
RNN on CUDA <https://github.com/TorchJD/torchjd/issues/220>`_, for instance. If you
70
- experience issues with ``backward`` try to use ``parallel_chunk_size=1`` to avoid relying on
+ experience issues with ``jac`` try to use ``parallel_chunk_size=1`` to avoid relying on
71
``torch.vmap``.
72
"""
73
0 commit comments