pytorch-benchmark (pypi): Easily benchmark PyTorch model FLOPs, latency, throughput, max allocated memory and energy consumption in one go.
Keywords: deep learning|pytorch|AI|benchmark|speed|energy|memory|deep-learning|flops|gpu|jetson|python|timing-analysis