I wish pytorch had something akin to the Tensorflow Profiler to easily visualize performance.