Recommended reading: Dive into Deep Learning — D2L: 2.4; 2.5; 5.3.

The Colab notebook contains the lecture code for Module 2 (backpropagation and PyTorch basics). Follow along by running the cells in order.