← all repositories

MrYxJ/calculate-flops.pytorch

A PyTorch-based tool that calculates theoretical FLOPs, MACs, and parameter counts for neural networks including transformers and LLMs.

943 stars Python LLMOps · Eval
calculate-flops.pytorch
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

Calflops computes floating-point operations, multiply-add operations, and parameter counts for PyTorch neural network models. It supports CNNs, RNNs, GCNs, transformers, and custom layers via torch.nn.function. It also provides a HuggingFace Space interface and a HuggingFace integration for directly analyzing models by name without downloading weights.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.