[juliangoldsmith]

Autograd hides the backward pass in that implementation, but this code spells it out explicitly.