#torch.compile

1 articles about torch.compile

Profiling in PyTorch (Part 2): From nn.Linear to a Fused MLP
AI Tools Jun 11, 2026

Profiling in PyTorch (Part 2): From nn.Linear to a Fused MLP

Part 2 of Hugging Face's PyTorch profiling series shows how to read torch.profiler traces from nn.Linear through compiled and hand-tuned MLP kernels, and what fuse decisions actually improve.