Profiling in PyTorch (Half 2): From nn.Linear to a Fused MLP
Within the first a part of this collection "Profiling in PyTorch", we used torch.add(torch.matmul(x, w), b) to discover ways to ...
Tag: Fused
The world's leading source for AI research, emerging technology, and the people building the future. Independent, rigorous, and always ahead.