torch-tb-profiler (pypi): PyTorch Profiler TensorBoard Plugin
Keywords: pytorch|tensorboard|profile|plugin