https://medium.com/@mujtaba.arhamsoft/exploring-deep-learning-with-tinygrad-training-and-testing-a-simple-neural-network-b8ef10e54cb2