tinygrad
last updated: Oct 20, 2023
https://github.com/geohot/tinygrad
Library from George Hotz focused on creating the smallest possible useful deep learning library.
Check the examples dir, for ex the mnist example that only uses 145 lines
# from the root dir of the repo
python examples/mnist_gan.py
on my computer each epoch takes about 3 minutes, so training the whole net (300 epochs) would take about 15 hours.
I saw in the source that there is a metal interface, but no idea if it's being used.