CentML/DeepView.Profile
🏙 Interactive performance profiling and debugging tool for PyTorch neural networks.
This tool helps machine learning engineers and researchers understand and fix performance bottlenecks in their PyTorch neural network training. It takes your PyTorch model's training code as input and provides detailed insights into runtime and memory usage on NVIDIA GPUs. The output helps identify where your model is spending the most time and consuming the most memory.
No commits in the last 6 months. Available on PyPI.
Use this if you are developing PyTorch models and need to identify why your training is slow or consuming too much GPU memory.
Not ideal if you are working with non-PyTorch frameworks, CPU-based training, or do not have access to an NVIDIA GPU.
Stars
64
Forks
11
Language
Python
License
Apache-2.0
Category
Last pushed
Jan 21, 2025
Commits (30d)
0
Dependencies
14
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/CentML/DeepView.Profile"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
lutzroeder/netron
Visualizer for neural network, deep learning and machine learning models
mert-kurttutan/torchview
torchview: visualize pytorch models
raghakot/keras-vis
Neural network visualization toolkit for keras
janosh/diagrams
Diagrams of concepts in physics/chemistry/ML
dvgodoy/deepreplay
Deep Replay - Generate visualizations as in my "Hyper-parameters in Action!" series!