eagomez2/moduleprofiler
Free open-source package to profile PyTorch models.
This tool helps machine learning engineers and researchers understand the computational cost and architecture of their PyTorch deep learning models. It takes a PyTorch model as input and provides detailed reports on parameters, operation counts, inference times, and input/output sizes for each component. This allows practitioners to design efficient models that meet specific performance and resource constraints.
Available on PyPI.
Use this if you are building PyTorch models and need to analyze their performance characteristics like memory footprint, speed, and complexity during development.
Not ideal if you are working with machine learning frameworks other than PyTorch or if you primarily need to profile the training phase of your model rather than inference.
Stars
10
Forks
—
Language
Python
License
CC-BY-SA-4.0
Category
Last pushed
Oct 17, 2025
Commits (30d)
0
Dependencies
4
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/eagomez2/moduleprofiler"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
deepspeedai/DeepSpeed
DeepSpeed is a deep learning optimization library that makes distributed training and inference...
helmholtz-analytics/heat
Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python
hpcaitech/ColossalAI
Making large AI models cheaper, faster and more accessible
horovod/horovod
Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
bsc-wdc/dislib
The Distributed Computing library for python implemented using PyCOMPSs programming model for HPC.