https://derekzhouai.github.io/posts/mlp-implementation-scratch/
Multilayer Perceptrons Implementation from Scratch (Pytorch) - Derek Zhou