nvidia-modelopt0.40.0
nvidia-modelopt0.40.0
Published
Nvidia Model Optimizer: a unified model optimization and deployment toolkit.
pip install nvidia-modelopt
Package Downloads
Authors
Project URLs
Requires Python
<3.13,>=3.10
Dependencies
- ninja
- numpy
- packaging
- pydantic
>=2.0 - nvidia-ml-py
>=12 - rich
- scipy
- tqdm
- pulp
- regex
- safetensors
- torch
>=2.6 - torchprofile
>=0.0.4 - cppimport
; extra == "onnx" - cupy-cuda12x
; (platform_machine != "aarch64" and platform_system != "Darwin") and extra == "onnx" - ml_dtypes
; extra == "onnx" - onnx-graphsurgeon
; extra == "onnx" - onnx
~=1.19.0; extra == "onnx" - onnxconverter-common
~=1.16.0; extra == "onnx" - onnxruntime
~=1.22.0; (platform_machine == "aarch64" or platform_system == "Darwin") and extra == "onnx" - onnxruntime-gpu
~=1.22.0; (platform_machine != "aarch64" and platform_system != "Darwin" and platform_system != "Windows") and extra == "onnx" - onnxruntime-directml
==1.20.0; platform_system == "Windows" and extra == "onnx" - onnxscript
; extra == "onnx" - onnxslim
>=0.1.76; extra == "onnx" - polygraphy
>=0.49.22; extra == "onnx" - accelerate
>=1.0.0; extra == "hf" - datasets
>=3.0.0; extra == "hf" - diffusers
>=0.32.2; extra == "hf" - huggingface_hub
>=0.24.0; extra == "hf" - peft
>=0.17.0; extra == "hf" - transformers
<5.0,>=4.53; extra == "hf" - deepspeed
>=0.9.6; (platform_system != "Darwin" and platform_system != "Windows") and extra == "hf" - bandit
[toml]==1.7.9; extra == "dev-lint" - mypy
==1.17.1; extra == "dev-lint" - pre-commit
==4.3.0; extra == "dev-lint" - ruff
==0.12.11; extra == "dev-lint" - coverage
; extra == "dev-test" - pytest
; extra == "dev-test" - pytest-cov
; extra == "dev-test" - pytest-instafail
; extra == "dev-test" - pytest-timeout
; extra == "dev-test" - timm
; extra == "dev-test" - torchvision
; extra == "dev-test" - torch-geometric
; extra == "dev-test" - tox
>4.18; extra == "dev-test" - tox-current-env
>=0.0.12; extra == "dev-test" - autodoc_pydantic
>=2.1.0; extra == "dev-docs" - sphinx
~=8.1.0; extra == "dev-docs" - sphinx-argparse
>=0.5.2; extra == "dev-docs" - sphinx-autobuild
>=2024.10.3; extra == "dev-docs" - sphinx-copybutton
>=0.5.2; extra == "dev-docs" - sphinx-inline-tabs
>=2023.4.21; extra == "dev-docs" - sphinx-rtd-theme
~=3.0.0; extra == "dev-docs" - sphinx-togglebutton
>=0.3.2; extra == "dev-docs" - cython
; extra == "dev-build" - setuptools
>=80; extra == "dev-build" - setuptools-scm
>=8; extra == "dev-build" - cppimport
; extra == "all" - cupy-cuda12x
; (platform_machine != "aarch64" and platform_system != "Darwin") and extra == "all" - ml_dtypes
; extra == "all" - onnx-graphsurgeon
; extra == "all" - onnx
~=1.19.0; extra == "all" - onnxconverter-common
~=1.16.0; extra == "all" - onnxruntime
~=1.22.0; (platform_machine == "aarch64" or platform_system == "Darwin") and extra == "all" - onnxruntime-gpu
~=1.22.0; (platform_machine != "aarch64" and platform_system != "Darwin" and platform_system != "Windows") and extra == "all" - onnxruntime-directml
==1.20.0; platform_system == "Windows" and extra == "all" - onnxscript
; extra == "all" - onnxslim
>=0.1.76; extra == "all" - polygraphy
>=0.49.22; extra == "all" - accelerate
>=1.0.0; extra == "all" - datasets
>=3.0.0; extra == "all" - diffusers
>=0.32.2; extra == "all" - huggingface_hub
>=0.24.0; extra == "all" - peft
>=0.17.0; extra == "all" - transformers
<5.0,>=4.53; extra == "all" - deepspeed
>=0.9.6; (platform_system != "Darwin" and platform_system != "Windows") and extra == "all" - cppimport
; extra == "dev" - cupy-cuda12x
; (platform_machine != "aarch64" and platform_system != "Darwin") and extra == "dev" - ml_dtypes
; extra == "dev" - onnx-graphsurgeon
; extra == "dev" - onnx
~=1.19.0; extra == "dev" - onnxconverter-common
~=1.16.0; extra == "dev" - onnxruntime
~=1.22.0; (platform_machine == "aarch64" or platform_system == "Darwin") and extra == "dev" - onnxruntime-gpu
~=1.22.0; (platform_machine != "aarch64" and platform_system != "Darwin" and platform_system != "Windows") and extra == "dev" - onnxruntime-directml
==1.20.0; platform_system == "Windows" and extra == "dev" - onnxscript
; extra == "dev" - onnxslim
>=0.1.76; extra == "dev" - polygraphy
>=0.49.22; extra == "dev" - accelerate
>=1.0.0; extra == "dev" - datasets
>=3.0.0; extra == "dev" - diffusers
>=0.32.2; extra == "dev" - huggingface_hub
>=0.24.0; extra == "dev" - peft
>=0.17.0; extra == "dev" - transformers
<5.0,>=4.53; extra == "dev" - deepspeed
>=0.9.6; (platform_system != "Darwin" and platform_system != "Windows") and extra == "dev" - bandit
[toml]==1.7.9; extra == "dev" - mypy
==1.17.1; extra == "dev" - pre-commit
==4.3.0; extra == "dev" - ruff
==0.12.11; extra == "dev" - coverage
; extra == "dev" - pytest
; extra == "dev" - pytest-cov
; extra == "dev" - pytest-instafail
; extra == "dev" - pytest-timeout
; extra == "dev" - timm
; extra == "dev" - torchvision
; extra == "dev" - torch-geometric
; extra == "dev" - tox
>4.18; extra == "dev" - tox-current-env
>=0.0.12; extra == "dev" - autodoc_pydantic
>=2.1.0; extra == "dev" - sphinx
~=8.1.0; extra == "dev" - sphinx-argparse
>=0.5.2; extra == "dev" - sphinx-autobuild
>=2024.10.3; extra == "dev" - sphinx-copybutton
>=0.5.2; extra == "dev" - sphinx-inline-tabs
>=2023.4.21; extra == "dev" - sphinx-rtd-theme
~=3.0.0; extra == "dev" - sphinx-togglebutton
>=0.3.2; extra == "dev" - cython
; extra == "dev" - setuptools
>=80; extra == "dev" - setuptools-scm
>=8; extra == "dev" - cppimport
; extra == "dev" - cupy-cuda12x
; (platform_machine != "aarch64" and platform_system != "Darwin") and extra == "dev" - ml_dtypes
; extra == "dev" - onnx-graphsurgeon
; extra == "dev" - onnx
~=1.19.0; extra == "dev" - onnxconverter-common
~=1.16.0; extra == "dev" - onnxruntime
~=1.22.0; (platform_machine == "aarch64" or platform_system == "Darwin") and extra == "dev" - onnxruntime-gpu
~=1.22.0; (platform_machine != "aarch64" and platform_system != "Darwin" and platform_system != "Windows") and extra == "dev" - onnxruntime-directml
==1.20.0; platform_system == "Windows" and extra == "dev" - onnxscript
; extra == "dev" - onnxslim
>=0.1.76; extra == "dev" - polygraphy
>=0.49.22; extra == "dev" - accelerate
>=1.0.0; extra == "dev" - datasets
>=3.0.0; extra == "dev" - diffusers
>=0.32.2; extra == "dev" - huggingface_hub
>=0.24.0; extra == "dev" - peft
>=0.17.0; extra == "dev" - transformers
<5.0,>=4.53; extra == "dev" - deepspeed
>=0.9.6; (platform_system != "Darwin" and platform_system != "Windows") and extra == "dev"
Checkout https://github.com/nvidia/Model-Optimizer for more information.