PyPI page
Home page
Author:
None
License:
Apache-2.0
Summary:
Predict GPU execution time & memory for PyTorch models — without running them.
Latest version:
0.2.0
Required dependencies:
joblib
|
numpy
|
pandas
|
scikit-learn
|
thop
|
xgboost
Optional dependencies:
black
|
blink-gpu
|
fastapi
|
httpx
|
lightgbm
|
matplotlib
|
mypy
|
optuna
|
plotly
|
pre-commit
|
pynvml
|
pytest
|
pytest-cov
|
python-multipart
|
ruff
|
seaborn
|
shap
|
streamlit
|
torch-geometric
|
uvicorn
Downloads last day:
8
Downloads last week:
27
Downloads last month:
112