Profiling in PyTorch (Part 2): From nn.Linear to a Fused MLP
Profiling in PyTorch (Part 2): From nn.Linear to a Fused MLP This is the second post of Profiling in PyTorch, a series where we slowly build the skill of reading profiler traces and use it to drive optimization: Profiling in PyTorch (Part...
Read more: https://huggingface.co/blog/torch-mlp-fusion
Scan with your phone
Scan with your phone