torchprofile (pypi): A general and accurate MACs / FLOPs profiler for PyTorch models
Keywords: profiler|pytorch