Metadata-Version: 2.4
Name: nvidia-modelopt
Version: 0.44.0rc1
Summary: Nvidia Model Optimizer: a unified model optimization and deployment toolkit.
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.14,>=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.6,>=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==1.17.1; extra == "dev-lint"
Requires-Dist: pre-commit==4.3.0; extra == "dev-lint"
Requires-Dist: ruff==0.12.11; extra == "dev-lint"
Provides-Extra: dev-docs
Requires-Dist: autodoc_pydantic>=2.1.0; extra == "dev-docs"
Requires-Dist: sphinx~=8.1.0; extra == "dev-docs"
Requires-Dist: sphinx-argparse>=0.5.2; 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-test
Requires-Dist: coverage[toml]>=7.13.0; extra == "dev-test"
Requires-Dist: nox; extra == "dev-test"
Requires-Dist: pytest; extra == "dev-test"
Requires-Dist: pytest-cov; extra == "dev-test"
Requires-Dist: pytest-instafail; extra == "dev-test"
Requires-Dist: pytest-timeout; 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.
