torch.compile vs eager for training a 468M Llama-style model on four GPUs: step time, memory and compile cost
Does torch.compile make a small transformer train faster? I rented four GPUs on Runcrate today (2026-10-29), an RTX 5090, an L40S, an RTX A6000 and an H100 PCIe, trained the same 468M-parameter Llama-style model in eager mode and compiled, with PyTorch's SDPA attention and with the plain "eager" at…
Read the full story at DEV Community — Machine Learning ↗
Timeline · 1 report
- 2026-10-29 03:24 · DEV Community — Machine Learning
torch.compile vs eager for training a 468M Llama-style model on four GPUs: step time, memory and compile cost