Thankfully pytorch does all of that calculus for us!