AI RESEARCH
[P] TraceML: wrap your PyTorch training step in single context manager and see what’s slowing training live
r/MachineLearning
•
End-summary Building TraceML, an open-source tool for PyTorch