reminder here is the github for Grokking Deep Learning. Easier to copy code from it (Atleast in my case from the O'rielly version) iamtrask/Grokking-Deep-Learning: this repository accompanies the book "Grokking Deep Learning" (github.com)