msaroufim/lit-llama

Simple (fast) transformer inference in PyTorch with torch.compile + lit-llama code

Python