Metadata-Version: 2.4
Name: nvidia-modelopt
Version: 0.29.0
Summary: Nvidia TensorRT Model Optimizer: a unified model optimization and deployment toolkit.
Home-page: https://github.com/NVIDIA/TensorRT-Model-Optimizer
Author: NVIDIA Corporation
License: Apache 2.0
Classifier: Programming Language :: Python :: 3
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.9,<3.13
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: nvidia-modelopt-core==0.29.0
Requires-Dist: ninja
Requires-Dist: numpy
Requires-Dist: packaging
Requires-Dist: pydantic>=2.0
Requires-Dist: rich
Requires-Dist: scipy
Requires-Dist: tqdm
Provides-Extra: deploy
Provides-Extra: onnx
Requires-Dist: cppimport; extra == "onnx"
Requires-Dist: cupy-cuda12x; (platform_machine != "aarch64" and platform_system != "Darwin") and extra == "onnx"
Requires-Dist: onnx; extra == "onnx"
Requires-Dist: onnxconverter-common; extra == "onnx"
Requires-Dist: onnx-graphsurgeon; extra == "onnx"
Requires-Dist: onnxruntime~=1.18.1; python_version < "3.10" and extra == "onnx"
Requires-Dist: onnxruntime~=1.20.1; (python_version >= "3.10" and (platform_machine == "aarch64" or platform_system == "Darwin")) and extra == "onnx"
Requires-Dist: onnxruntime-gpu~=1.20.1; (python_version >= "3.10" and platform_machine != "aarch64" and platform_system != "Darwin") and extra == "onnx"
Requires-Dist: onnxsim; (python_version < "3.12" and platform_machine != "aarch64") and extra == "onnx"
Provides-Extra: torch
Requires-Dist: pulp; extra == "torch"
Requires-Dist: pynvml>=11.5.0; extra == "torch"
Requires-Dist: regex; extra == "torch"
Requires-Dist: safetensors; extra == "torch"
Requires-Dist: torch>=2.4; extra == "torch"
Requires-Dist: torchprofile>=0.0.4; extra == "torch"
Requires-Dist: torchvision; extra == "torch"
Provides-Extra: hf
Requires-Dist: accelerate>=1.0.0; extra == "hf"
Requires-Dist: datasets>=3.0.0; extra == "hf"
Requires-Dist: diffusers>=0.32.2; extra == "hf"
Requires-Dist: huggingface_hub>=0.24.0; extra == "hf"
Requires-Dist: peft>=0.12.0; extra == "hf"
Requires-Dist: transformers<4.52.0,>=4.48.0; extra == "hf"
Provides-Extra: dev-lint
Requires-Dist: bandit[toml]==1.7.9; extra == "dev-lint"
Requires-Dist: mypy==1.15.0; extra == "dev-lint"
Requires-Dist: pre-commit==4.2.0; extra == "dev-lint"
Requires-Dist: ruff==0.11.6; extra == "dev-lint"
Provides-Extra: dev-test
Requires-Dist: coverage; extra == "dev-test"
Requires-Dist: onnxscript; extra == "dev-test"
Requires-Dist: pytest; extra == "dev-test"
Requires-Dist: pytest-asyncio; extra == "dev-test"
Requires-Dist: pytest-cov; extra == "dev-test"
Requires-Dist: pytest-timeout; extra == "dev-test"
Requires-Dist: timm; extra == "dev-test"
Requires-Dist: toml; extra == "dev-test"
Requires-Dist: tox; extra == "dev-test"
Requires-Dist: tox-current-env>=0.0.12; extra == "dev-test"
Provides-Extra: dev-docs
Requires-Dist: autodoc_pydantic>=2.1.0; extra == "dev-docs"
Requires-Dist: ipython; extra == "dev-docs"
Requires-Dist: ipywidgets; extra == "dev-docs"
Requires-Dist: nbsphinx>=0.9.3; extra == "dev-docs"
Requires-Dist: pypandoc; extra == "dev-docs"
Requires-Dist: sphinx~=7.2.0; extra == "dev-docs"
Requires-Dist: snowballstemmer<3; extra == "dev-docs"
Requires-Dist: sphinx-autobuild>=2024.10.3; extra == "dev-docs"
Requires-Dist: sphinx-copybutton>=0.5.2; extra == "dev-docs"
Requires-Dist: sphinx-inline-tabs>=2023.4.21; extra == "dev-docs"
Requires-Dist: sphinx-rtd-theme~=3.0.0; extra == "dev-docs"
Requires-Dist: sphinx-togglebutton>=0.3.2; extra == "dev-docs"
Provides-Extra: dev-build
Requires-Dist: cython; extra == "dev-build"
Requires-Dist: setuptools>=67.8.0; extra == "dev-build"
Requires-Dist: setuptools_scm>=7.1.0; extra == "dev-build"
Requires-Dist: twine; extra == "dev-build"
Provides-Extra: all
Requires-Dist: cppimport; extra == "all"
Requires-Dist: cupy-cuda12x; (platform_machine != "aarch64" and platform_system != "Darwin") and extra == "all"
Requires-Dist: onnx; extra == "all"
Requires-Dist: onnxconverter-common; extra == "all"
Requires-Dist: onnx-graphsurgeon; extra == "all"
Requires-Dist: onnxruntime~=1.18.1; python_version < "3.10" and extra == "all"
Requires-Dist: onnxruntime~=1.20.1; (python_version >= "3.10" and (platform_machine == "aarch64" or platform_system == "Darwin")) and extra == "all"
Requires-Dist: onnxruntime-gpu~=1.20.1; (python_version >= "3.10" and platform_machine != "aarch64" and platform_system != "Darwin") and extra == "all"
Requires-Dist: onnxsim; (python_version < "3.12" and platform_machine != "aarch64") and extra == "all"
Requires-Dist: pulp; extra == "all"
Requires-Dist: pynvml>=11.5.0; extra == "all"
Requires-Dist: regex; extra == "all"
Requires-Dist: safetensors; extra == "all"
Requires-Dist: torch>=2.4; extra == "all"
Requires-Dist: torchprofile>=0.0.4; extra == "all"
Requires-Dist: torchvision; extra == "all"
Requires-Dist: accelerate>=1.0.0; extra == "all"
Requires-Dist: datasets>=3.0.0; extra == "all"
Requires-Dist: diffusers>=0.32.2; extra == "all"
Requires-Dist: huggingface_hub>=0.24.0; extra == "all"
Requires-Dist: peft>=0.12.0; extra == "all"
Requires-Dist: transformers<4.52.0,>=4.48.0; extra == "all"
Provides-Extra: dev
Requires-Dist: cppimport; extra == "dev"
Requires-Dist: cupy-cuda12x; (platform_machine != "aarch64" and platform_system != "Darwin") and extra == "dev"
Requires-Dist: onnx; extra == "dev"
Requires-Dist: onnxconverter-common; extra == "dev"
Requires-Dist: onnx-graphsurgeon; extra == "dev"
Requires-Dist: onnxruntime~=1.18.1; python_version < "3.10" and extra == "dev"
Requires-Dist: onnxruntime~=1.20.1; (python_version >= "3.10" and (platform_machine == "aarch64" or platform_system == "Darwin")) and extra == "dev"
Requires-Dist: onnxruntime-gpu~=1.20.1; (python_version >= "3.10" and platform_machine != "aarch64" and platform_system != "Darwin") and extra == "dev"
Requires-Dist: onnxsim; (python_version < "3.12" and platform_machine != "aarch64") and extra == "dev"
Requires-Dist: pulp; extra == "dev"
Requires-Dist: pynvml>=11.5.0; extra == "dev"
Requires-Dist: regex; extra == "dev"
Requires-Dist: safetensors; extra == "dev"
Requires-Dist: torch>=2.4; extra == "dev"
Requires-Dist: torchprofile>=0.0.4; extra == "dev"
Requires-Dist: torchvision; extra == "dev"
Requires-Dist: accelerate>=1.0.0; extra == "dev"
Requires-Dist: datasets>=3.0.0; extra == "dev"
Requires-Dist: diffusers>=0.32.2; extra == "dev"
Requires-Dist: huggingface_hub>=0.24.0; extra == "dev"
Requires-Dist: peft>=0.12.0; extra == "dev"
Requires-Dist: transformers<4.52.0,>=4.48.0; extra == "dev"
Requires-Dist: bandit[toml]==1.7.9; extra == "dev"
Requires-Dist: mypy==1.15.0; extra == "dev"
Requires-Dist: pre-commit==4.2.0; extra == "dev"
Requires-Dist: ruff==0.11.6; extra == "dev"
Requires-Dist: coverage; extra == "dev"
Requires-Dist: onnxscript; extra == "dev"
Requires-Dist: pytest; extra == "dev"
Requires-Dist: pytest-asyncio; extra == "dev"
Requires-Dist: pytest-cov; extra == "dev"
Requires-Dist: pytest-timeout; extra == "dev"
Requires-Dist: timm; extra == "dev"
Requires-Dist: toml; extra == "dev"
Requires-Dist: tox; extra == "dev"
Requires-Dist: tox-current-env>=0.0.12; extra == "dev"
Requires-Dist: autodoc_pydantic>=2.1.0; extra == "dev"
Requires-Dist: ipython; extra == "dev"
Requires-Dist: ipywidgets; extra == "dev"
Requires-Dist: nbsphinx>=0.9.3; extra == "dev"
Requires-Dist: pypandoc; extra == "dev"
Requires-Dist: sphinx~=7.2.0; extra == "dev"
Requires-Dist: snowballstemmer<3; extra == "dev"
Requires-Dist: sphinx-autobuild>=2024.10.3; extra == "dev"
Requires-Dist: sphinx-copybutton>=0.5.2; extra == "dev"
Requires-Dist: sphinx-inline-tabs>=2023.4.21; extra == "dev"
Requires-Dist: sphinx-rtd-theme~=3.0.0; extra == "dev"
Requires-Dist: sphinx-togglebutton>=0.3.2; extra == "dev"
Requires-Dist: cython; extra == "dev"
Requires-Dist: setuptools>=67.8.0; extra == "dev"
Requires-Dist: setuptools_scm>=7.1.0; extra == "dev"
Requires-Dist: twine; extra == "dev"
Requires-Dist: cppimport; extra == "dev"
Requires-Dist: cupy-cuda12x; (platform_machine != "aarch64" and platform_system != "Darwin") and extra == "dev"
Requires-Dist: onnx; extra == "dev"
Requires-Dist: onnxconverter-common; extra == "dev"
Requires-Dist: onnx-graphsurgeon; extra == "dev"
Requires-Dist: onnxruntime~=1.18.1; python_version < "3.10" and extra == "dev"
Requires-Dist: onnxruntime~=1.20.1; (python_version >= "3.10" and (platform_machine == "aarch64" or platform_system == "Darwin")) and extra == "dev"
Requires-Dist: onnxruntime-gpu~=1.20.1; (python_version >= "3.10" and platform_machine != "aarch64" and platform_system != "Darwin") and extra == "dev"
Requires-Dist: onnxsim; (python_version < "3.12" and platform_machine != "aarch64") and extra == "dev"
Requires-Dist: pulp; extra == "dev"
Requires-Dist: pynvml>=11.5.0; extra == "dev"
Requires-Dist: regex; extra == "dev"
Requires-Dist: safetensors; extra == "dev"
Requires-Dist: torch>=2.4; extra == "dev"
Requires-Dist: torchprofile>=0.0.4; extra == "dev"
Requires-Dist: torchvision; extra == "dev"
Requires-Dist: accelerate>=1.0.0; extra == "dev"
Requires-Dist: datasets>=3.0.0; extra == "dev"
Requires-Dist: diffusers>=0.32.2; extra == "dev"
Requires-Dist: huggingface_hub>=0.24.0; extra == "dev"
Requires-Dist: peft>=0.12.0; extra == "dev"
Requires-Dist: transformers<4.52.0,>=4.48.0; extra == "dev"
Dynamic: author
Dynamic: classifier
Dynamic: description
Dynamic: description-content-type
Dynamic: home-page
Dynamic: license
Dynamic: license-file
Dynamic: provides-extra
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary

Checkout https://github.com/nvidia/TensorRT-Model-Optimizer for more information.
