Metadata-Version: 2.4
Name: nvidia-modelopt
Version: 0.46.0rc0
Summary: Nvidia Model Optimizer: A unified library of SOTA model optimization techniques like quantization, pruning, Neural Architecture Search (NAS), distillation, speculative decoding, etc. It compresses deep learning models for downstream deployment frameworks like TensorRT-LLM, TensorRT, vLLM, etc. to optimize inference speed.
Author: NVIDIA Corporation
License-Expression: Apache-2.0
Project-URL: Homepage, https://github.com/NVIDIA/Model-Optimizer
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: <3.15,>=3.10
Description-Content-Type: text/markdown
License-File: LICENSE_HEADER
Requires-Dist: ninja
Requires-Dist: numpy
Requires-Dist: nvidia-ml-py>=12
Requires-Dist: packaging
Requires-Dist: setuptools>=80
Requires-Dist: torch>=2.8
Requires-Dist: tqdm
Requires-Dist: PyYAML>=6.0
Requires-Dist: omegaconf>=2.3.0
Requires-Dist: pulp<4.0
Requires-Dist: pydantic>=2.0
Requires-Dist: regex
Requires-Dist: rich
Requires-Dist: safetensors
Requires-Dist: scipy
Provides-Extra: onnx
Requires-Dist: cppimport; extra == "onnx"
Requires-Dist: cupy-cuda12x; (platform_machine != "aarch64" and platform_system != "Darwin") and extra == "onnx"
Requires-Dist: lief; extra == "onnx"
Requires-Dist: ml_dtypes; extra == "onnx"
Requires-Dist: onnx-graphsurgeon>=0.6.1; extra == "onnx"
Requires-Dist: onnx~=1.21.0; extra == "onnx"
Requires-Dist: onnxconverter-common~=1.16.0; extra == "onnx"
Requires-Dist: onnxruntime-gpu==1.22.0; platform_system == "Windows" and extra == "onnx"
Requires-Dist: onnxruntime~=1.22.0; (python_version <= "3.10" and (platform_machine == "aarch64" or platform_system == "Darwin")) and extra == "onnx"
Requires-Dist: onnxruntime-gpu~=1.22.0; (python_version <= "3.10" and platform_machine != "aarch64" and platform_system != "Darwin" and platform_system != "Windows") and extra == "onnx"
Requires-Dist: onnxruntime~=1.24.2; (python_version > "3.10" and (platform_machine == "aarch64" or platform_system == "Darwin")) and extra == "onnx"
Requires-Dist: onnxruntime-gpu~=1.24.2; (python_version > "3.10" and platform_machine != "aarch64" and platform_system != "Darwin" and platform_system != "Windows") and extra == "onnx"
Requires-Dist: onnxscript; extra == "onnx"
Requires-Dist: onnxslim>=0.1.76; extra == "onnx"
Requires-Dist: polygraphy>=0.49.22; extra == "onnx"
Provides-Extra: hf
Requires-Dist: accelerate>=1.0.0; extra == "hf"
Requires-Dist: datasets>=3.0.0; extra == "hf"
Requires-Dist: deepspeed>=0.9.6; (platform_system != "Darwin" and platform_system != "Windows") and extra == "hf"
Requires-Dist: diffusers>=0.32.2; extra == "hf"
Requires-Dist: huggingface_hub>=0.24.0; extra == "hf"
Requires-Dist: nltk; extra == "hf"
Requires-Dist: peft>=0.17.0; extra == "hf"
Requires-Dist: sentencepiece>=0.2.1; extra == "hf"
Requires-Dist: tiktoken; extra == "hf"
Requires-Dist: transformers<5.13,>=4.56; extra == "hf"
Requires-Dist: wonderwords; extra == "hf"
Provides-Extra: puzzletron
Requires-Dist: fire; extra == "puzzletron"
Requires-Dist: hydra-core==1.3.2; extra == "puzzletron"
Requires-Dist: immutabledict; extra == "puzzletron"
Requires-Dist: lru-dict; extra == "puzzletron"
Requires-Dist: pandas; extra == "puzzletron"
Requires-Dist: typeguard; extra == "puzzletron"
Provides-Extra: dev-lint
Requires-Dist: bandit[toml]==1.7.9; extra == "dev-lint"
Requires-Dist: mypy==2.1.0; extra == "dev-lint"
Requires-Dist: pre-commit==4.6.0; extra == "dev-lint"
Requires-Dist: ruff==0.15.20; extra == "dev-lint"
Provides-Extra: dev-docs
Requires-Dist: autodoc_pydantic~=2.2.0; python_version >= "3.12" and extra == "dev-docs"
Requires-Dist: sphinx~=9.1.0; python_version >= "3.12" and extra == "dev-docs"
Requires-Dist: sphinx-argparse~=0.6.0; python_version >= "3.12" and extra == "dev-docs"
Requires-Dist: sphinx-autobuild==2025.8.25; python_version >= "3.12" and extra == "dev-docs"
Requires-Dist: sphinx-copybutton~=0.5.2; python_version >= "3.12" and extra == "dev-docs"
Requires-Dist: sphinx-inline-tabs==2025.12.21.14; python_version >= "3.12" and extra == "dev-docs"
Requires-Dist: sphinx-rtd-theme~=3.1.0; python_version >= "3.12" and extra == "dev-docs"
Requires-Dist: sphinx-togglebutton~=0.4.0; python_version >= "3.12" and extra == "dev-docs"
Provides-Extra: dev-test
Requires-Dist: coverage[toml]~=7.14.0; extra == "dev-test"
Requires-Dist: nox; extra == "dev-test"
Requires-Dist: pytest~=9.1.0; extra == "dev-test"
Requires-Dist: pytest-cov~=7.1.0; extra == "dev-test"
Requires-Dist: pytest-instafail==0.5.0; extra == "dev-test"
Requires-Dist: pytest-timeout~=2.4.0; extra == "dev-test"
Requires-Dist: uv; extra == "dev-test"
Requires-Dist: timm; extra == "dev-test"
Requires-Dist: torchprofile==0.0.4; extra == "dev-test"
Requires-Dist: torchvision; extra == "dev-test"
Requires-Dist: torch-geometric; extra == "dev-test"
Provides-Extra: all
Requires-Dist: nvidia-modelopt[hf,onnx,puzzletron]; extra == "all"
Provides-Extra: dev
Requires-Dist: nvidia-modelopt[all,dev-docs,dev-lint,dev-test]; extra == "dev"
Dynamic: license-file

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