Flops Python Packages

ultralytics-thop

Profile PyTorch models for FLOPs and parameters, helping to evaluate computational efficiency and memory usage.

keras-flops

FLOPs calculator with tf.profiler for neural network architecture written in tensorflow 2.2+ (tf.keras)

torchtoolbox

🛠 Toolbox to extend PyTorch functionalities

pytorch-benchmark

Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption

calibrax

Unified benchmarking and profiling framework for the JAX scientific ML ecosystem. Timing, GPU/energy monitoring, FLOPS counting, roofline analysis, statistical testing, regression detection, and CI integration.

pthflops

Estimate/count FLOPS for a given neural network using pytorch

torchscan

Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)

flopco-pytorch

FLOPs and other statistics COunter for Pytorch neural networks

ppma

PaddlePaddle Model Analysis.

chinchilla

A toolkit for scaling law research ⚖

floppy-tracker

A hardware-agnostic profiler for tracking the FLOPs and BOPs of Machine and Deep Learning algorithms.

flopco-keras

FLOPs and other statistics COunter for TF.keras neural networks

floppy-lib

A hardware-agnostic python library to monitor the computational cost of Machine and Deep Learning algorithms.