Short name describing what triggered the graph break
torch.cond: differing branch outputs
Values or code snippet captured at the break point
true_spec: {true_spec.treespec}, false_spec: {false_spec.treespec}, same_spec: {same_spec}
Explanation of why the graph break was triggered
Expected branches to return the same pytree structure.
Hints on how to resolve the graph break