Wheelodex » Projects » optimum-nvidia

optimum-nvidia

View on PyPI — Reverse Dependencies (0)

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

Wheel Details

Project: optimum-nvidia
Version: 0.1.0b7
Filename: optimum_nvidia-0.1.0b7-py3-none-any.whl
Download: [link]
Size: 77453
MD5: 2f5c43acb07c58f64804e7c1fd25e484
SHA256: 959fd7de505f70b1e66430ed81ec7e0d43b28b53225b9affb4c8948b664f6fa8
Uploaded: 2024-05-24 13:10:27 +0000

dist-info

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

METADATA

Metadata-Version: 2.1
Name: optimum-nvidia
Version: 0.1.0b7
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.25)
Requires-Dist: datasets (>=2.14.0)
Requires-Dist: huggingface-hub (>=0.22.0)
Requires-Dist: hf-transfer
Requires-Dist: mpmath (==1.3.0)
Requires-Dist: numpy (<2.0.0,>=1.26.0)
Requires-Dist: onnx (>=1.12.0)
Requires-Dist: optimum (>=1.13.0)
Requires-Dist: setuptools
Requires-Dist: tensorrt-llm (==0.9.0)
Requires-Dist: torch (<=2.3.0a,>=2.2.0a)
Requires-Dist: transformers (>=4.38.2)
Requires-Dist: pynvml
Requires-Dist: mock; extra == "test"
Requires-Dist: pytest; extra == "test"
Requires-Dist: pytest-xdist; extra == "test"
Requires-Dist: psutil; extra == "test"
Requires-Dist: parameterized; extra == "test"
Requires-Dist: datasets; extra == "test"
Requires-Dist: safetensors; extra == "test"
Requires-Dist: mock; extra == "tests"
Requires-Dist: pytest; extra == "tests"
Requires-Dist: pytest-xdist; extra == "tests"
Requires-Dist: psutil; extra == "tests"
Requires-Dist: parameterized; extra == "tests"
Requires-Dist: datasets; extra == "tests"
Requires-Dist: safetensors; extra == "tests"
Requires-Dist: soundfile; extra == "tests"
Requires-Dist: librosa; extra == "tests"
Provides-Extra: test
Provides-Extra: tests
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 5169 characters]

WHEEL

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

RECORD

Path Digest Size
optimum/__init__.py sha256=EITJmPSWi90s6OYNW8Xs_Fe-0Am7lk7ZFmoPCqPqrLQ 646
optimum/nvidia/__init__.py sha256=WCGL3NsCcg7eQCN-2WmgGkZNCMDjFR_Ewk_h2w9gfLA 853
optimum/nvidia/config.py sha256=i83ZOMkkRT7OiaBuDpMbbkJ2yaO6exC9GKbHSZVSq4Y 7692
optimum/nvidia/errors.py sha256=mQ6OJCNL_EW8VZkISwx7khtKEGQXA5xReKCLCK_Y38Q 2027
optimum/nvidia/hub.py sha256=ppVuv4T2QzUxWiQ8CXAjs2WlVIkAKaWDQJsVCKXwJHE 21839
optimum/nvidia/logging.py sha256=wZJ0_mvlBZY2p03bVFZg975Yef0Bx_ig9VWa1gBtuqU 987
optimum/nvidia/runtime.py sha256=zjtqjZTV9XloGzc8KZ4-QT53CBbeRYynkTQf8Y00b7I 13954
optimum/nvidia/version.py sha256=xarswUevY84DmlLero1rMSTK4qXzeMj5HOiwE2IPIgo 754
optimum/nvidia/builder/__init__.py sha256=EeGnAjDBDR_IGVbU-vPzTd1-hD_y065NMiK1S6JSu3s 686
optimum/nvidia/builder/config.py sha256=iD1NIJhC02jG71YOsIX9ov4GExdzc-DkTJDWF7Fmlxs 9713
optimum/nvidia/builder/local.py sha256=Ahjed2DHBd9_E_n6Y1FLYz0ZEIgERH1TcB7W6bndpcM 5473
optimum/nvidia/generation/logits_process.py sha256=GAcG30JkR3KRIMUktwpC3w5CwltQkXKvF2azLa0e5FA 1448
optimum/nvidia/lang/__init__.py sha256=_M9fcKkBvjyxlkjo_0-a0kKl3IMOEiGiqcd174eT-Tc 3808
optimum/nvidia/models/__init__.py sha256=VbqCKHH_cnR6oiqIe1gpqh6UYfa2wMwlZPlxy9_vITA 728
optimum/nvidia/models/auto.py sha256=5-VsvvkBxhaVEFbXccIRmFigDvTxb1sghfKdXY9he18 2687
optimum/nvidia/models/base.py sha256=zQvH6_sXcGtxSl3b52362ZQAMB-kuSsis4gdfTfu5cY 1365
optimum/nvidia/models/gemma.py sha256=A61AHVDvTBJTq5Ri0Cz45wXSBEj6ZBcXbWsAUVztbLI 4725
optimum/nvidia/models/llama.py sha256=tPshOnq3lOVPFJzNU7DCEtduNJeOJh2yjd3zWDrqXCQ 4778
optimum/nvidia/models/mistral.py sha256=Fhrz1pHFL2xJsKPIU9Zw2Xbfm8GyB7qbqSxqY1_vUuo 4755
optimum/nvidia/models/mixtral.py sha256=13HhBM1AKI1LiWwd8NSl738WuuLxbbzQjSk6bC9-Blw 4889
optimum/nvidia/models/whisper.py sha256=MqqZDrul7golP-e1vzB4ldLUZLDilPAx6l2jWgWUYYI 51158
optimum/nvidia/pipelines/__init__.py sha256=Ec02i0530k5zPdud51Zzkvez_owLRTIAt-aMzc7Mj84 5052
optimum/nvidia/pipelines/base.py sha256=nljvL_tIdyhZHiAc1WHB24bXz2OOdHSVXfICa7Dznp8 792
optimum/nvidia/pipelines/text_generation.py sha256=AqX0ooK8CErsRcmIXVvtXSU6leqcbiteKnvUyOSO6q0 9226
optimum/nvidia/quantization/__init__.py sha256=n8ZVfg4Xasb9YYWPkinYTIxPGZoBoIbcH-zIR5JHJQg 779
optimum/nvidia/quantization/datasets.py sha256=Rww4AulPK2t8SWelfPbU6iZejPVVLNYJt3dAInk1vJA 10814
optimum/nvidia/quantization/ammo/__init__.py sha256=qLtvlxx02xzB7IJdO6u-IzAVFbJzJQhlSunHkVy705w 171
optimum/nvidia/quantization/ammo/config.py sha256=X1NODuMWBlLhbliEOLA5WoL6XeGc44EVBndADYVnWcs 5810
optimum/nvidia/quantization/ammo/quantizer.py sha256=Me33mFMt7PtfsSKFKKawa6jUQsKNMRRNLvBqLgdsZQc 4907
optimum/nvidia/utils/__init__.py sha256=JYVKrkuXfnb7_UG9c681FcEZ8fePAnBDeTEMwUP-otA 1422
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=dyf_iL91y447YxyXAo9p8iQR8gcKAc0xvshqtrxpAt4 2655
optimum/nvidia/utils/nvml.py sha256=YRNhoast7mlwZ7hYYy-Bhq0YchCGQ5bLpDt5AuZ6BFE 3098
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=J3eQ8gcd8MQ8K6mpDCcSPZpq-Bgpt7WTfRA_zx91f6U 973
optimum/nvidia/utils/tests/utils.py sha256=DpT-6Fc8w_NUk4y35tGeU1U_-KwPwfVPZbLdm4ZGnzI 2084
optimum_nvidia-0.1.0b7.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
optimum_nvidia-0.1.0b7.dist-info/METADATA sha256=-KePjNVgxtGFlagFAUmv0bNsWlq2v8ueCtj9NjwmPY8 7493
optimum_nvidia-0.1.0b7.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
optimum_nvidia-0.1.0b7.dist-info/dependency_links.txt sha256=zsVzSy_1j9zQ3hSWuhg3E1YODR492Jyjb4VQ7BLJs3o 24
optimum_nvidia-0.1.0b7.dist-info/top_level.txt sha256=rc1rV-uPZnV1Ek7hCwh56pvMCKcip65JbHRRWs8Yqu8 8
optimum_nvidia-0.1.0b7.dist-info/RECORD — —

top_level.txt

optimum
https://pypi.nvidia.com