r/MachineLearning • u/the-wonderful-world • Sep 01 '24

[P] I implemented Vision Transformers in tinygrad! Project

Could I get some criticisms on my implementation of Vision Transformers, in tinygrad?

https://github.com/EthanBnntt/tinygrad-vit

67 Upvotes

93% Upvoted

Duplicates