nvidia-modelopt 0.46.0rc0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (3) hide show
  1. nvidia_modelopt-0.46.0rc0/LICENSE_HEADER +14 -0
  2. nvidia_modelopt-0.46.0rc0/PKG-INFO +101 -0
  3. nvidia_modelopt-0.46.0rc0/pyproject.toml +10 -0
nvidia_modelopt-0.46.0rc0/LICENSE_HEADER ADDED
@@ -0,0 +1,14 @@
1
+ SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2
+ SPDX-License-Identifier: Apache-2.0
3
+
4
+ Licensed under the Apache License, Version 2.0 (the "License");
5
+ you may not use this file except in compliance with the License.
6
+ You may obtain a copy of the License at
7
+
8
+ http://www.apache.org/licenses/LICENSE-2.0
9
+
10
+ Unless required by applicable law or agreed to in writing, software
11
+ distributed under the License is distributed on an "AS IS" BASIS,
12
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
+ See the License for the specific language governing permissions and
14
+ limitations under the License.
nvidia_modelopt-0.46.0rc0/PKG-INFO ADDED
@@ -0,0 +1,101 @@
1
+ Metadata-Version: 2.4
2
+ Name: nvidia-modelopt
3
+ Version: 0.46.0rc0
4
+ 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.
5
+ Author: NVIDIA Corporation
6
+ License-Expression: Apache-2.0
7
+ Project-URL: Homepage, https://github.com/NVIDIA/Model-Optimizer
8
+ Classifier: Programming Language :: Python :: 3
9
+ Classifier: Programming Language :: Python :: 3.10
10
+ Classifier: Programming Language :: Python :: 3.11
11
+ Classifier: Programming Language :: Python :: 3.12
12
+ Classifier: Programming Language :: Python :: 3.13
13
+ Classifier: Intended Audience :: Developers
14
+ Classifier: Intended Audience :: Science/Research
15
+ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
16
+ Requires-Python: <3.15,>=3.10
17
+ Description-Content-Type: text/markdown
18
+ License-File: LICENSE_HEADER
19
+ Requires-Dist: ninja
20
+ Requires-Dist: numpy
21
+ Requires-Dist: nvidia-ml-py>=12
22
+ Requires-Dist: packaging
23
+ Requires-Dist: setuptools>=80
24
+ Requires-Dist: torch>=2.8
25
+ Requires-Dist: tqdm
26
+ Requires-Dist: PyYAML>=6.0
27
+ Requires-Dist: omegaconf>=2.3.0
28
+ Requires-Dist: pulp<4.0
29
+ Requires-Dist: pydantic>=2.0
30
+ Requires-Dist: regex
31
+ Requires-Dist: rich
32
+ Requires-Dist: safetensors
33
+ Requires-Dist: scipy
34
+ Provides-Extra: onnx
35
+ Requires-Dist: cppimport; extra == "onnx"
36
+ Requires-Dist: cupy-cuda12x; (platform_machine != "aarch64" and platform_system != "Darwin") and extra == "onnx"
37
+ Requires-Dist: lief; extra == "onnx"
38
+ Requires-Dist: ml_dtypes; extra == "onnx"
39
+ Requires-Dist: onnx-graphsurgeon>=0.6.1; extra == "onnx"
40
+ Requires-Dist: onnx~=1.21.0; extra == "onnx"
41
+ Requires-Dist: onnxconverter-common~=1.16.0; extra == "onnx"
42
+ Requires-Dist: onnxruntime-gpu==1.22.0; platform_system == "Windows" and extra == "onnx"
43
+ Requires-Dist: onnxruntime~=1.22.0; (python_version <= "3.10" and (platform_machine == "aarch64" or platform_system == "Darwin")) and extra == "onnx"
44
+ 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"
45
+ Requires-Dist: onnxruntime~=1.24.2; (python_version > "3.10" and (platform_machine == "aarch64" or platform_system == "Darwin")) and extra == "onnx"
46
+ 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"
47
+ Requires-Dist: onnxscript; extra == "onnx"
48
+ Requires-Dist: onnxslim>=0.1.76; extra == "onnx"
49
+ Requires-Dist: polygraphy>=0.49.22; extra == "onnx"
50
+ Provides-Extra: hf
51
+ Requires-Dist: accelerate>=1.0.0; extra == "hf"
52
+ Requires-Dist: datasets>=3.0.0; extra == "hf"
53
+ Requires-Dist: deepspeed>=0.9.6; (platform_system != "Darwin" and platform_system != "Windows") and extra == "hf"
54
+ Requires-Dist: diffusers>=0.32.2; extra == "hf"
55
+ Requires-Dist: huggingface_hub>=0.24.0; extra == "hf"
56
+ Requires-Dist: nltk; extra == "hf"
57
+ Requires-Dist: peft>=0.17.0; extra == "hf"
58
+ Requires-Dist: sentencepiece>=0.2.1; extra == "hf"
59
+ Requires-Dist: tiktoken; extra == "hf"
60
+ Requires-Dist: transformers<5.13,>=4.56; extra == "hf"
61
+ Requires-Dist: wonderwords; extra == "hf"
62
+ Provides-Extra: puzzletron
63
+ Requires-Dist: fire; extra == "puzzletron"
64
+ Requires-Dist: hydra-core==1.3.2; extra == "puzzletron"
65
+ Requires-Dist: immutabledict; extra == "puzzletron"
66
+ Requires-Dist: lru-dict; extra == "puzzletron"
67
+ Requires-Dist: pandas; extra == "puzzletron"
68
+ Requires-Dist: typeguard; extra == "puzzletron"
69
+ Provides-Extra: dev-lint
70
+ Requires-Dist: bandit[toml]==1.7.9; extra == "dev-lint"
71
+ Requires-Dist: mypy==2.1.0; extra == "dev-lint"
72
+ Requires-Dist: pre-commit==4.6.0; extra == "dev-lint"
73
+ Requires-Dist: ruff==0.15.20; extra == "dev-lint"
74
+ Provides-Extra: dev-docs
75
+ Requires-Dist: autodoc_pydantic~=2.2.0; python_version >= "3.12" and extra == "dev-docs"
76
+ Requires-Dist: sphinx~=9.1.0; python_version >= "3.12" and extra == "dev-docs"
77
+ Requires-Dist: sphinx-argparse~=0.6.0; python_version >= "3.12" and extra == "dev-docs"
78
+ Requires-Dist: sphinx-autobuild==2025.8.25; python_version >= "3.12" and extra == "dev-docs"
79
+ Requires-Dist: sphinx-copybutton~=0.5.2; python_version >= "3.12" and extra == "dev-docs"
80
+ Requires-Dist: sphinx-inline-tabs==2025.12.21.14; python_version >= "3.12" and extra == "dev-docs"
81
+ Requires-Dist: sphinx-rtd-theme~=3.1.0; python_version >= "3.12" and extra == "dev-docs"
82
+ Requires-Dist: sphinx-togglebutton~=0.4.0; python_version >= "3.12" and extra == "dev-docs"
83
+ Provides-Extra: dev-test
84
+ Requires-Dist: coverage[toml]~=7.14.0; extra == "dev-test"
85
+ Requires-Dist: nox; extra == "dev-test"
86
+ Requires-Dist: pytest~=9.1.0; extra == "dev-test"
87
+ Requires-Dist: pytest-cov~=7.1.0; extra == "dev-test"
88
+ Requires-Dist: pytest-instafail==0.5.0; extra == "dev-test"
89
+ Requires-Dist: pytest-timeout~=2.4.0; extra == "dev-test"
90
+ Requires-Dist: uv; extra == "dev-test"
91
+ Requires-Dist: timm; extra == "dev-test"
92
+ Requires-Dist: torchprofile==0.0.4; extra == "dev-test"
93
+ Requires-Dist: torchvision; extra == "dev-test"
94
+ Requires-Dist: torch-geometric; extra == "dev-test"
95
+ Provides-Extra: all
96
+ Requires-Dist: nvidia-modelopt[hf,onnx,puzzletron]; extra == "all"
97
+ Provides-Extra: dev
98
+ Requires-Dist: nvidia-modelopt[all,dev-docs,dev-lint,dev-test]; extra == "dev"
99
+ Dynamic: license-file
100
+
101
+ Checkout https://github.com/nvidia/Model-Optimizer for more information.
nvidia_modelopt-0.46.0rc0/pyproject.toml ADDED
@@ -0,0 +1,10 @@
1
+ # SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2
+ # SPDX-License-Identifier: Apache-2
3
+
4
+ [build-system]
5
+ requires = ["wheel-stub"]
6
+ build-backend = "wheel_stub.buildapi"
7
+
8
+ [tool.wheel_stub]
9
+ index_url = "https://pypi.nvidia.com/"
10
+ include_cuda_debuginfo = true