Coding LLaMA 3 from scratch in PyTorch