rewrote the autograd engine in Pytorch mostly cos I was curious as to how it worked but ended up making a pretty smoother version of it
you might have to zoom in but it was able to perform the gradients on this network pretty smoothly without any issues