Swall0w/torchstat

Health v2 · maintenance only

Adoption not part of the score

1500 stars · 138 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

torchstat is a lightweight neural network analyzer for PyTorch that reports parameter counts, FLOPs, multiply-adds, and memory usage per layer. It can be used as a CLI tool or imported as a Python module to profile nn.Module models.

Use cases

• count total parameters in a PyTorch model
• estimate FLOPs and multiply-adds of a neural network
• measure memory usage of model layers
• profile layer-by-layer compute cost of a network
• debug PyTorch model architecture statistics

When to choose

• you need quick parameter/FLOPs/memory stats for a PyTorch nn.Module
• you want a simple CLI or one-line stat() call for model analysis

When to avoid

• you need support for ops like Dropout2d or newer PyTorch modules
• you need actively maintained tooling with modern PyTorch compatibility
• you need runtime profiling rather than theoretical estimates

Facets

library · maturity maintenance

benchmarking developer-tools cli deep-learning machine-learning developer-tools performance python cli cross-platform pytorch model-analysis flops memory-usage neural-networks profiling

Member repositories

For agents

markdown · JSON · MCP: product_card(name="Swall0w/torchstat")