Wheelodex » Projects » optimum-nvidia

optimum-nvidia

View on PyPI — Reverse Dependencies (0)

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

Wheel Details

Project: optimum-nvidia
Version: 0.1.0b6
Filename: optimum_nvidia-0.1.0b6-py3-none-any.whl
Download: [link]
Size: 73735
MD5: c2a80dd896777db0e48c581cb72aeaff
SHA256: 6aa8fb61be830a84df3a9bf3d08da174df69dc435b8589480770d9895354376e
Uploaded: 2024-04-11 21:13:36 +0000

dist-info

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

METADATA

Metadata-Version: 2.1
Name: optimum-nvidia
Version: 0.1.0b6
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 (>=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.dev2024031900)
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: 5099 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=K2XBR2w3DinqnXqm7yt7eOOFfFg4kx7474mmU1rQbcs 7297
optimum/nvidia/errors.py sha256=_DEom41367f-bjT8EPhloTCPYNe5al5KTejYJQahZGU 1855
optimum/nvidia/hub.py sha256=VxsdlNCuvbwrj64DzPitpFdp5yVfeGwcM4gcoPgUvhQ 19797
optimum/nvidia/logging.py sha256=wZJ0_mvlBZY2p03bVFZg975Yef0Bx_ig9VWa1gBtuqU 987
optimum/nvidia/runtime.py sha256=4ohnweP9TiZQd68iSuG1jqNzui7mW0EdZX_0LbHf9jQ 13945
optimum/nvidia/version.py sha256=SBK84hTc9d7WHjzDY8BGv3EEqTQwiwQFz5K7waBPO-U 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=wPHLB9-JebP-IW4ynik1AQHMK-vfCHAy81m_Dvxsj-4 5180
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=CJv2NTcOTXQwbbYQiMFtaPzhLoJo2HchXt5SAVyPuqA 2612
optimum/nvidia/models/base.py sha256=zQvH6_sXcGtxSl3b52362ZQAMB-kuSsis4gdfTfu5cY 1365
optimum/nvidia/models/gemma.py sha256=ds0i2aulABXW2B6S0zL-yWehtHEYUXI4yzPuIsR0BdE 4447
optimum/nvidia/models/llama.py sha256=8vBiCnVgqVZaPmNMCmB2jytVVeeupJl_FPK_TGO6U_0 4426
optimum/nvidia/models/mistral.py sha256=_dh9uG2rZK8rVTcUJ_WjrThLEBvExG4DcB0jmSoRBcI 4483
optimum/nvidia/models/whisper.py sha256=I3eTlPiElDYQoOGsMmC0EbJ3D8pD7pvYM9WE9zYekqk 50629
optimum/nvidia/pipelines/__init__.py sha256=m-gjQ0Sfx-6zDBtrlrGYfY9i45CJiD0Ly1xB1tF0JPY 4968
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=sx2DZlc-BM6jYZrKkMZ7YUFC6dB9rnwpIQgiWUJovyI 4131
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=TzTaU0jpaH6gt6rdV3sSiVTltd6RevYxURH25fhM1Sw 694
optimum/nvidia/utils/tests/utils.py sha256=obBcEnWNmL0lqze3IKksBrcSzWaYdkAkchb1C3LD81k 1972
optimum_nvidia-0.1.0b6.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
optimum_nvidia-0.1.0b6.dist-info/METADATA sha256=QXHu2lhcUVJ2KHO_qJzRGIKN__GiA5I4VccGkXQFsdQ 7430
optimum_nvidia-0.1.0b6.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
optimum_nvidia-0.1.0b6.dist-info/dependency_links.txt sha256=zsVzSy_1j9zQ3hSWuhg3E1YODR492Jyjb4VQ7BLJs3o 24
optimum_nvidia-0.1.0b6.dist-info/top_level.txt sha256=rc1rV-uPZnV1Ek7hCwh56pvMCKcip65JbHRRWs8Yqu8 8
optimum_nvidia-0.1.0b6.dist-info/RECORD — —

top_level.txt

optimum
https://pypi.nvidia.com