Simple PyTorch profiler that combines DeepSpeed Flops Profiler and TorchInfo
Primary LanguagePythonMIT LicenseMIT