This is the code for "PyTorch in 5 Minutes" by Siraj Raval on Youtube - View it on GitHub