nvidia-modelopt 0.44.0

Author: NVIDIA Corporation

Requires Python: <3.15,>=3.10

nvidia_modelopt-0.44.0-py3-none-any.whl (1.5MiB)

Dependencies:

ninja

numpy

nvidia-ml-py (>=12)

packaging

setuptools (>=80)

torch (>=2.8)

tqdm

PyYAML (>=6.0)

omegaconf (>=2.3.0)

pulp (<4.0)

pydantic (>=2.0)

regex

rich

safetensors

scipy

cppimport extra == "onnx"

cupy-cuda12x (platform_machine != "aarch64" and platform_system != "Darwin") and extra == "onnx"

lief extra == "onnx"

ml_dtypes extra == "onnx"

onnx-graphsurgeon extra == "onnx" (>=0.6.1)

onnx extra == "onnx" (~=1.21.0)

onnxconverter-common extra == "onnx" (~=1.16.0)

onnxruntime-gpu platform_system == "Windows" and extra == "onnx" (==1.22.0)

onnxruntime (python_version <= "3.10" and (platform_machine == "aarch64" or platform_system == "Darwin")) and extra == "onnx" (~=1.22.0)

onnxruntime-gpu (python_version <= "3.10" and platform_machine != "aarch64") and extra == "onnx" (~=1.22.0)

onnxruntime (python_version > "3.10" and (platform_machine == "aarch64" or platform_system == "Darwin")) and extra == "onnx" (~=1.24.2)

onnxruntime-gpu (python_version > "3.10" and platform_machine != "aarch64") and extra == "onnx" (~=1.24.2)

onnxscript extra == "onnx"

onnxslim extra == "onnx" (>=0.1.76)

polygraphy extra == "onnx" (>=0.49.22)

accelerate extra == "hf" (>=1.0.0)

datasets extra == "hf" (>=3.0.0)

deepspeed (platform_system != "Darwin" and platform_system != "Windows") and extra == "hf" (<0.19,>=0.9.6)

diffusers extra == "hf" (>=0.32.2)

huggingface_hub extra == "hf" (>=0.24.0)

nltk extra == "hf"

peft extra == "hf" (>=0.17.0)

sentencepiece extra == "hf" (>=0.2.1)

tiktoken extra == "hf"

transformers extra == "hf" (<5.6,>=4.56)

wonderwords extra == "hf"

fire extra == "puzzletron"

hydra-core extra == "puzzletron" (==1.3.2)

immutabledict extra == "puzzletron"

lru-dict extra == "puzzletron"

pandas extra == "puzzletron"

typeguard extra == "puzzletron"

bandit[toml] extra == "dev-lint" (==1.7.9)

mypy extra == "dev-lint" (==1.17.1)

pre-commit extra == "dev-lint" (==4.3.0)

ruff extra == "dev-lint" (==0.12.11)

autodoc_pydantic extra == "dev-docs" (>=2.1.0)

sphinx extra == "dev-docs" (~=8.1.0)

sphinx-argparse extra == "dev-docs" (>=0.5.2)

sphinx-autobuild extra == "dev-docs" (>=2024.10.3)

sphinx-copybutton extra == "dev-docs" (>=0.5.2)

sphinx-inline-tabs extra == "dev-docs" (>=2023.4.21)

sphinx-rtd-theme extra == "dev-docs" (~=3.0.0)

sphinx-togglebutton extra == "dev-docs" (>=0.3.2)

coverage[toml] extra == "dev-test" (>=7.13.0)

nox extra == "dev-test"

pytest extra == "dev-test"

pytest-cov extra == "dev-test"

pytest-instafail extra == "dev-test"

pytest-timeout extra == "dev-test"

uv extra == "dev-test"

timm extra == "dev-test"

torchprofile extra == "dev-test" (==0.0.4)

torchvision extra == "dev-test"

torch-geometric extra == "dev-test"

nvidia-modelopt[onnx,puzzletron,hf] extra == "all"

nvidia-modelopt[dev-test,dev-lint,all,dev-docs] extra == "dev"