Skip to content

If tinygrad wasn't small enough for you...

License

Notifications You must be signed in to change notification settings

mbilalsardar/teenygrad

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

teenygrad is <1000 line MNIST trainer

It shares 90% of its code with tinygrad,
so understanding teenygrad is a good step to understanding tinygrad.

While it supports most of tinygrad's functionality,
the extra 3k lines in tinygrad get you speed and diverse backend support.

tensor.py and mlops.py are both tinygrad's and teenygrad's frontend.
lazy.py is a replacement for all of tinygrad's backend.

The only dependency of teenygrad is numpy (well, and tqdm)

Usage:
pip install numpy tqdm
PYTHONPATH="." python mnist.py

About

If tinygrad wasn't small enough for you...

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%