tinygrad by geohot, also linked on this thread, has similar properties good for learning - it's a couple hundred LoC to integrate LLaMA support https://github.

tinygrad by geohot, also linked on this thread, has similar properties good for learning - it's a couple hundred LoC to integrate LLaMA support https://github.com/geohot/tinygrad/tree/llama https://github.com/geohot/tinygrad/tree/llama