PyPI page
Home page
Author:
None
License:
Apache 2.0
Summary:
TraceML: Lightweight runtime bottleneck diagnostics for PyTorch training.
Latest version:
0.2.14
Required dependencies:
ipython
|
ipywidgets
|
msgspec
|
nicegui
|
numpy
|
nvidia-ml-py
|
pandas
|
plotly
|
psutil
|
rich
|
scikit-learn
Optional dependencies:
accelerate
|
black
|
codespell
|
coverage
|
datasets
|
isort
|
lightning
|
mkdocs
|
mkdocs-autorefs
|
mkdocs-include-markdown-plugin
|
mkdocs-material
|
mkdocstrings
|
nbstripout
|
pre-commit
|
pytest
|
pytest-cov
|
ruff
|
torch
|
torchvision
|
transformers
Downloads last day:
25
Downloads last week:
252
Downloads last month:
1,164