Flops Python Packages

ultralytics-thop

PyTorch model profiler for computing MACs and parameter counts to measure deep learning model complexity.

calibrax

JAX benchmarking, profiling and evaluation metrics for Flax NNX: a registry of pure-function metrics (regression, classification, calibration, uncertainty, forecasting, generative, image, text, audio, graph, fairness), XLA FLOP counting, roofline analysis, GPU and energy monitoring, regression detection, publication exports, W&B and MLflow.

keras-flops

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

floppy-tracker

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

torchscan

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

pytorch-benchmark

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

torchtoolbox

🛠 Toolbox to extend PyTorch functionalities

pthflops

Estimate/count FLOPS for a given neural network using pytorch

chinchilla

A toolkit for scaling law research ⚖

ppma

【一个小工具】用最短的代码完成对模型的分析，包含 ImageNet Val、FLOPs、Params、Throuthput、CAM 等

flopco-pytorch

FLOPs and other statistics COunter for Pytorch neural networks

flopco-keras

FLOPs and other statistics COunter for tf.keras neural networks