Click pytorch_mlp.ipynb link to view the file.