This repos contains a small implementation of neural network architecture that mimics the pytorch. It uses computation graph to compute gradient based on the autodiff. I used the repos by Andre Karapathy to modle mine.
you can run .ipynb file to run the neural nets