Wheelodex » Projects » optimum-nvidia

optimum-nvidia

View on PyPI — Reverse Dependencies (0)

0.1.0b9 optimum_nvidia-0.1.0b9-py3-none-any.whl

Wheel Details

Project: optimum-nvidia
Version: 0.1.0b9
Filename: optimum_nvidia-0.1.0b9-py3-none-any.whl
Download: [link]
Size: 57212
MD5: b5099cfc2d9e71ff96f800c2e68738bf
SHA256: 79a42efc4e3f28ea6e60a012e925f3bd0ec76acd41bb09c256df0c94a6a03e0a
Uploaded: 2025-01-21 15:17:34 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt · entry_points.txt · dependency_links.txt

METADATA

Metadata-Version: 2.2
Name: optimum-nvidia
Version: 0.1.0b9
Summary: Optimum Nvidia is the interface between the Hugging Face Transformers and NVIDIA GPUs. "
Author: HuggingFace Inc. Machine Learning Optimization Team
Author-Email: "HuggingFace Inc. Machine Learning Optimization Team" <hardware[at]huggingface.co>
Home-Page: https://huggingface.co/hardware/nvidia
Project-Url: Homepage, https://huggingface.co/hardware/nvidia
Project-Url: Repository, https://github.com/huggingface/optimum-nvidia
Project-Url: Issues, https://github.com/huggingface/optimum-nvidia/issues
License: Apache/2.0
Keywords: transformers,neural-network,inference,nvidia,tensorrt,ampere,hopper
Classifier: Development Status :: 2 - Pre-Alpha
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Education
Classifier: Intended Audience :: Science/Research
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3.10
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10
Requires-Dist: accelerate (>=0.26)
Requires-Dist: datasets (>=2.14.0)
Requires-Dist: huggingface-hub (>=0.24.0)
Requires-Dist: hf-transfer (==0.1.6)
Requires-Dist: mpi4py (<4.0.0)
Requires-Dist: mpmath (==1.3.0)
Requires-Dist: numpy (<2.0.0,>=1.26.0)
Requires-Dist: onnx (>=1.17.0)
Requires-Dist: optimum (>=1.21.0)
Requires-Dist: setuptools
Requires-Dist: tensorrt-llm (==0.16.0)
Requires-Dist: torch (<=2.6.0a,>=2.4.0a)
Requires-Dist: transformers (>=4.45.1)
Requires-Dist: pynvml
Requires-Dist: mock; extra == "tests"
Requires-Dist: pytest; extra == "tests"
Requires-Dist: pytest-console-scripts; extra == "tests"
Requires-Dist: pytest-xdist; extra == "tests"
Requires-Dist: psutil; extra == "tests"
Requires-Dist: parameterized; extra == "tests"
Provides-Extra: tests
Description-Content-Type: text/markdown
Dynamic: author
Dynamic: home-page
License-File: LICENSE
[Description omitted; length: 5167 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.8.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
optimum/commands/env.py sha256=BfmVzPzKUyoD2E2UfSXm9Wm8Xhp_5W-Mc3sNFWG-m38 1874
optimum/nvidia/__init__.py sha256=CbPP0fDsPMDYcds8exDglfoRHc-VwYKpGOlN7H3qLi8 880
optimum/nvidia/errors.py sha256=IEnlxlX0CRkHIZSdEta0KZ-W3tuAXHx4GX9mjvE8ErI 1977
optimum/nvidia/hub.py sha256=EM95d-OnkHgDF1fCBV57w4vNCm44ejzTtoE_ATIldQU 17528
optimum/nvidia/logging.py sha256=RIAqGZL1dOE4RV951r-dufNEftklTW8JpYaY4pd9YwM 1059
optimum/nvidia/runtime.py sha256=uQel-88vFbPuRLIU-UY6UwZQKaccpAbcqIUWsCFdfLk 7072
optimum/nvidia/version.py sha256=dZhg8C2D6RlvSEjtWEWfYrmBvg9xfav0R0z_XYNfyzc 742
optimum/nvidia/compression/__init__.py sha256=rLX3TuxGb7OxbiN4lWn-LRUEeceMepbY-4hbtwUYNQU 141
optimum/nvidia/compression/modelopt.py sha256=9I-aYa9oAId9hG02HRnuLmQ2CXCh-D69uyh8BMoCB5Q 3846
optimum/nvidia/export/__init__.py sha256=-fFV_bwdVXr--Y8cruMMM5B0UVmeHWaHu_JHI1qiyjM 327
optimum/nvidia/export/cli.py sha256=40aNbgGGWICB2ljgk33fOxvuZ6Do9vedF68VhTBJSKw 2353
optimum/nvidia/export/config.py sha256=xDzwtP8RV9exCx800R_bmUloXpj6GlheAzZ4w36nFCU 7579
optimum/nvidia/export/converter.py sha256=1tx3xhNASi7W3gMTS2okzzh9Tj-lgeXNeGKKperPxeY 5297
optimum/nvidia/export/workspace.py sha256=ExSKlgXWWWi_lVosk-mBG-5rWyUnTsghe0CiQEIvINI 2084
optimum/nvidia/generation/logits_process.py sha256=GAcG30JkR3KRIMUktwpC3w5CwltQkXKvF2azLa0e5FA 1448
optimum/nvidia/lang/__init__.py sha256=NimTKfQpjYeARSRGld-x36n81VL6i9o27Gh1UTAcWWo 3813
optimum/nvidia/models/__init__.py sha256=hoeGTG7E5bc2HygjnwI2P0qHBKiRkuAb4tH_BJduUiQ 768
optimum/nvidia/models/auto.py sha256=Keqt_46TInYBZBg7XUK2BW-DQ_9QMHzL0WPfs_5iE58 3026
optimum/nvidia/models/base.py sha256=gYaUWSxjHGkIJJ9mF1OMSh-1WYMqy7Tt-qi0lzz6ruc 1131
optimum/nvidia/models/gemma.py sha256=7XKo_Kqj2ATb2wArtFGgVyYl9qtLS6Ucu41W5H2_k2k 1311
optimum/nvidia/models/llama.py sha256=Um9oAo8bSylWLmVBIrV7OZP3c-4P5bpbU4y6M8q4hvU 1138
optimum/nvidia/models/mistral.py sha256=P5tc6G_ESnEXMXwuHMafShNnWnh3UClDuvEwp9-e1-A 1218
optimum/nvidia/models/mixtral.py sha256=Boluz8n9x5jGfAtqugD2S1SaaE6NIN3KpuHMszr0mXY 1218
optimum/nvidia/models/whisper.py sha256=t6goL4sVIbvoHCo_hi6L7eamJ1HLMLTZKJlesDK2BfU 1392
optimum/nvidia/pipelines/__init__.py sha256=8W_OteNYOfqQbIUmxwdYl8FgHAYEA9_Edkp1aggFcW0 5053
optimum/nvidia/pipelines/base.py sha256=nljvL_tIdyhZHiAc1WHB24bXz2OOdHSVXfICa7Dznp8 792
optimum/nvidia/pipelines/text_generation.py sha256=AqX0ooK8CErsRcmIXVvtXSU6leqcbiteKnvUyOSO6q0 9226
optimum/nvidia/subpackage/__init__.py sha256=i4RJNG8Ba-LlhhERcQsUg7Yr1lGpPsyl4hlKDkz5Okw 631
optimum/nvidia/subpackage/commands/__init__.py sha256=kH6_E74HAC4VXmy7e2j6qdpZQSW6CLEjnECodxiAA6A 683
optimum/nvidia/subpackage/commands/env.py sha256=mFpbJjW4OYb4jjubr0LJT45UcF-t09hWfToB9HfXYY8 229
optimum/nvidia/subpackage/commands/export.py sha256=6ygU97HNkWSW67gz8-8yNjDyoUORJhl80g9gDW04re0 3651
optimum/nvidia/utils/__init__.py sha256=j9CSTQdmdRJkogHJTX6WmJrJdr5cRXTeCGW5qMBTtdY 1452
optimum/nvidia/utils/cli.py sha256=dVROAejiVkPptp-DwjEcXLdI4X7zH3-JYhHzlcBNsPE 4169
optimum/nvidia/utils/constants.py sha256=7N3WGryHoJFZy0kMNKlWuVymNVoU4kxWbNw1he2mz9g 1040
optimum/nvidia/utils/env.py sha256=JdWLIdla_61Ss06peiFXqGOP9IKleaTc7Fwi5jNHRB0 1337
optimum/nvidia/utils/hub.py sha256=fN9-yj4mUBUnKoO8Hd4-0PmyVA4FXIb9ABmbgCXE65c 3655
optimum/nvidia/utils/import_utils.py sha256=7z-EeSqtHgtGzGVcQF8EDQSShp9CCKGsBYgsTjUUnNo 129
optimum/nvidia/utils/nvml.py sha256=qYkrQBFpUpYzwHyi4CtBgEY-fouPvQYsqlVpebunlts 3602
optimum/nvidia/utils/offload.py sha256=S5_PPI7Yk-n7erLeOsA-4aafwhvIzzAYcbwRhlUJxXE 2023
optimum/nvidia/utils/onnx.py sha256=c_lw4sLUp8xh78wZgmzT2mpj6iP_zpoXnE_ORQqvz8k 2798
optimum/nvidia/utils/patching.py sha256=x1Pwfg4y8wmZ-0w1Eu58O919ERjvfzVwUzqewav7eGI 1438
optimum/nvidia/utils/tests/__init__.py sha256=RogDZNr32W0z3-GVxKa9Qzn8QSV2sfO6W5oxyoWA64g 840
optimum/nvidia/utils/tests/assertions.py sha256=PCi1Rbkjy5_rKefpvK2nggj5Asm-6d6RmyoXB4N2okk 977
optimum/nvidia/utils/tests/utils.py sha256=DpT-6Fc8w_NUk4y35tGeU1U_-KwPwfVPZbLdm4ZGnzI 2084
optimum_nvidia-0.1.0b9.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
optimum_nvidia-0.1.0b9.dist-info/METADATA sha256=Mh1RKtksPKN7_JuHIicWZ0FgWCl6Uv7xNUuFDMZvSzY 7109
optimum_nvidia-0.1.0b9.dist-info/WHEEL sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8 91
optimum_nvidia-0.1.0b9.dist-info/dependency_links.txt sha256=zsVzSy_1j9zQ3hSWuhg3E1YODR492Jyjb4VQ7BLJs3o 24
optimum_nvidia-0.1.0b9.dist-info/entry_points.txt sha256=1__lv2jpNhLb5ebCqqbs8FFPhOfi9_3nH1dP1lvvZ9s 66
optimum_nvidia-0.1.0b9.dist-info/top_level.txt sha256=rc1rV-uPZnV1Ek7hCwh56pvMCKcip65JbHRRWs8Yqu8 8
optimum_nvidia-0.1.0b9.dist-info/RECORD — —

top_level.txt

optimum

entry_points.txt

[console_scripts]
optimum-cli = optimum.commands.optimum_cli:main
https://pypi.nvidia.com