Automatic Differentiation

Gradient of the attention op