PyTorch layer-by-layer model profiler - View it on GitHub