Okay fuck I'm finally sold on ChatGPT. It can't do any complex mathematical coding like back-propagation functions in PyTorch but it can take all the effort out of the grunt work. This script would've taken me up to an hour to write and I was able to get it done in 2 minutes