torch_profiler. Simple PyTorch profiler that combines DeepSpeed Flops Profiler and TorchInfo

github.com/ruipeterpan/torch_profiler
by @ruipeterpan

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.