GitHub repository: mit-han-lab/torchprofile
Primary language: Python
Description: A more general and accurate MACs / FLOPs profiler for PyTorch models.