CS336 Lecture 2 Notes - PyTorch, Resource Accounting