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

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.