PyTorch comes with a rich set of tools for building machine learning models. This video will explain the Module class, which is the parent class of all PyTorch models; the Parameter class, a special kind of tensor representing learning weights in a Module; an overview of some of the common neural network layer types; and a sampling of other layer types and functions that are useful for building ML models.
Download the Building Models notebook here: 4 - Building Models in