The forward pass matches the reference implementation.
Next
Write the backward pass.