Wheelodex » Projects » tensor-parallel

tensor-parallel

View on PyPI — Reverse Dependencies (4)

2.0.0 tensor_parallel-2.0.0-py3-none-any.whl

Wheel Details

Project: tensor-parallel
Version: 2.0.0
Filename: tensor_parallel-2.0.0-py3-none-any.whl
Download: [link]
Size: 37954
MD5: 2d5e6f572334d58d132dd1755f1f55e0
SHA256: 8a8154d5a4eb8934595f2987b44c2e8069e7441f4325e8a5a3b85216a9b7b1b2
Uploaded: 2023-08-06 14:21:47 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: tensor-parallel
Version: 2.0.0
Summary: Automatically shard your large model between multiple GPUs, works without torch.distributed
Author: Andrei Panferov and Yaroslav Lisnyak
Author-Email: yalisnyak[at]nes.com
Home-Page: https://github.com/BlackSamorez/tensor_parallel
Project-Url: Bug Tracker, https://github.com/BlackSamorez/tensor_parallel/issues
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.7
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Topic :: Scientific/Engineering
Classifier: Topic :: Scientific/Engineering :: Mathematics
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development
Classifier: Topic :: Software Development :: Libraries
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Python: >=3.7
Requires-Dist: torch (>=1.11)
Requires-Dist: transformers (>=4.20.1)
Requires-Dist: pytest (==6.2.5); extra == "dev"
Requires-Dist: pytest-forked; extra == "dev"
Requires-Dist: pytest-asyncio (==0.16.0); extra == "dev"
Requires-Dist: accelerate (==0.20.3); extra == "dev"
Requires-Dist: black (==22.3.0); extra == "dev"
Requires-Dist: isort (==5.10.1); extra == "dev"
Requires-Dist: psutil; extra == "dev"
Requires-Dist: peft (==0.3.0); extra == "dev"
Requires-Dist: einops (==0.6.1); extra == "dev"
Provides-Extra: dev
Description-Content-Type: text/markdown
License-File: LICENCE
[Description omitted; length: 8359 characters]

WHEEL

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

RECORD

Path Digest Size
tensor_parallel/__init__.py sha256=Z8fmansbpLPCZd4flnEHZWS2cw50_7c9twprmHtS2yU 623
tensor_parallel/autoconfig.py sha256=KLz8qY8bwoJI8gfBukA9SAyKLoJib3GGT5KR-SOKjc4 4402
tensor_parallel/aux_actions.py sha256=tcLKFXHjCN_pruV2f6gWkgpYEDftHO0jr3MzMtRb0QM 2051
tensor_parallel/communications.py sha256=bY6gWTrxHh_y73SdhzWiFcvYRpAGb8b0fBYgEGyiavk 8541
tensor_parallel/config.py sha256=tUMXh6jZNL5p2gUOplVLvbNg5EoxMw08J5Xc3owhzeg 8056
tensor_parallel/cross_device_ops.py sha256=HgJI9FVnynCDxoy-4YVnNibxE-zfIV9sGSb7zLdTKFE 4504
tensor_parallel/dispatch.py sha256=GBQvar4makamKpWxdlF7HVg6hLEu6IajG-2cdgH8T58 5521
tensor_parallel/factory.py sha256=lGMvSYYPPoERsSKUEEco3zCzTZaVkPYAQknHNejfDxA 3665
tensor_parallel/imports.py sha256=dZgEFq9SQxBntT3LZEQnQQy2yctDcQh4EhO8oAi5hxk 220
tensor_parallel/legacy.py sha256=tnuY8DLzxbDtvHbuxOw-5RyxCvChCcHypv6y5Ss3_M0 618
tensor_parallel/per_device_tensors.py sha256=tdKIL1IXfdsnArS05DwqsnbBJO0fBXgwpAfTIRlqEz4 648
tensor_parallel/pretrained_model.py sha256=5m41djyzpSGZl5HLkjT0G9axCf3w-WW40ET2PEvSJWk 6818
tensor_parallel/shard.py sha256=o2vUHpdKu17-eHPVQXl3L24AQ13Ss6MYj7q2mT3Wqtc 8095
tensor_parallel/sharding.py sha256=ciKGbMLLh26sww-4xc1JE1w2-M1r_FTHmEO2f6QAxhE 9014
tensor_parallel/slicing_configs.py sha256=0YpF3S0UBCzUjj6WQIqR_54KTeHr3P_n07Zt02J__lo 22130
tensor_parallel/state_actions.py sha256=6v_YMVWKxMWPfb07bgr3rf9INHZOpDOIVa22RibW31k 4331
tensor_parallel/tensor_parallel.py sha256=reDYzNbXJGv1zCzggZ0nQJc7cBpzU0_Vx7uYtBF9kLM 14056
tensor_parallel/utils.py sha256=k78JUvWqOmvd9ndHhpq2r4K6gP2YRDVuUDb9PdzZHuk 4447
tensor_parallel/wrapper.py sha256=aa6qu46oxuLszLGotGVC3lXh7o4hsbCwx7mL_i1QRK4 3034
tensor_parallel-2.0.0.dist-info/LICENCE sha256=Iwk0HfpElXxl8bPKeI68zn0RAAhhScI4fxfwJ1HEQh4 1097
tensor_parallel-2.0.0.dist-info/METADATA sha256=v8hQ3hLnbL1xskrU2txxaiMVnDY4QS2WQNsrAR4q2AM 10117
tensor_parallel-2.0.0.dist-info/WHEEL sha256=5sUXSg9e4bi7lTLOHcm6QEYwO5TIF1TNbTSVFVjcJcc 92
tensor_parallel-2.0.0.dist-info/top_level.txt sha256=J9SOeFA9wTUrq7bl4bdYhKXsU6l7tbDyXzeLgUCZ8mw 16
tensor_parallel-2.0.0.dist-info/RECORD — —

top_level.txt

tensor_parallel