I'm really excited about this - profiling PyTorch training can be a total black box, but this could change everything. Just think about how much time this could save on tuning models!