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)

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.

torchtoolbox

🛠 Toolbox to extend PyTorch functionalities

pytorch-benchmark

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

pthflops

Estimate FLOPs of neural networks

torchscan

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

floppy-tracker

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

flopco-pytorch

FLOPs and other statistics COunter for Pytorch neural networks

ppma

PaddlePaddle Model Analysis.

chinchilla

A toolkit for scaling law research ⚖

flopco-keras

FLOPs and other statistics COunter for tf.keras neural networks

floppy-lib

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