PyPI page
Home page
Author:
NVIDIA Corporation
License:
Apache 2.0
Summary:
Nvidia Model Optimizer: a unified model optimization and deployment toolkit.
Latest version:
0.40.0
Required dependencies:
cupy-cuda12x
|
deepspeed
|
ninja
|
numpy
|
nvidia-ml-py
|
onnxruntime
|
onnxruntime-directml
|
onnxruntime-gpu
|
packaging
|
pulp
|
pydantic
|
regex
|
rich
|
safetensors
|
scipy
|
torch
|
torchprofile
|
tqdm
Optional dependencies:
accelerate
|
autodoc_pydantic
|
bandit
|
coverage
|
cppimport
|
cython
|
datasets
|
diffusers
|
huggingface_hub
|
ml_dtypes
|
mypy
|
onnx
|
onnx-graphsurgeon
|
onnxconverter-common
|
onnxscript
|
onnxslim
|
peft
|
polygraphy
|
pre-commit
|
pytest
|
pytest-cov
|
pytest-instafail
|
pytest-timeout
|
ruff
|
setuptools
|
setuptools-scm
|
sphinx
|
sphinx-argparse
|
sphinx-autobuild
|
sphinx-copybutton
|
sphinx-inline-tabs
|
sphinx-rtd-theme
|
sphinx-togglebutton
|
timm
|
torch-geometric
|
torchvision
|
tox
|
tox-current-env
|
transformers
Downloads last day:
5,609
Downloads last week:
44,858
Downloads last month:
181,891