Short name describing what triggered the graph break
autograd.Function.apply: _materialize_non_diff_grads mutation
Values or code snippet captured at the break point
Explanation of why the graph break was triggered
Mutations to autograd.Function.ctx._materialize_non_diff_grads are not supported.
Hints on how to resolve the graph break