Wheelodex » Projects » pai-easydist » pai_easydist-0.1.0-py3-none-any.whl

pai-easydist

View on PyPI — Reverse Dependencies (0)

0.1.0 pai_easydist-0.1.0-py3-none-any.whl

Wheel Details

Project: pai-easydist
Version: 0.1.0
Filename: pai_easydist-0.1.0-py3-none-any.whl
Download: [link]
Size: 93491
MD5: f56dfeb2d880ba56012b3d4c449b25a2
SHA256: 747585927a82742039be8b30d11c4d36ca1b121f495a448e07668b880d679fa7
Uploaded: 2023-09-08 08:04:22 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: pai-easydist
Version: 0.1.0
Summary: Efficient Automatic Training System for Super-Large Models
Author: Shenggan Cheng
Author-Email: shenggan.c[at]u.nus.edu
Home-Page: https://github.com/alibaba/easydist
Requires-Dist: pyyaml
Requires-Dist: rich
Requires-Dist: mip (>=1.13)
Requires-Dist: jax[cuda11_pip]; extra == "jax"
Requires-Dist: flax; extra == "jax"
Requires-Dist: torch; extra == "torch"
Requires-Dist: torchvision; extra == "torch"
Provides-Extra: jax
Provides-Extra: torch
Description-Content-Type: text/markdown
License-File: LICENSE
License-File: NOTICE
[Description omitted; length: 3036 characters]

WHEEL

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

RECORD

Path Digest Size
easydist/__init__.py sha256=QsJzfmKtgotNNGrGYjHk_bWnuF5IUSvgiolo8z4PIuQ 1458
easydist/config.py sha256=UVgYniBaEV4gJrJ5V3dgFLni5wN109d9zoB-ERcdON8 1557
easydist/version.py sha256=5Tu5fotGonIfoaz4bWYUuj9dkuGFwAl_nsfGB6ON8r0 774
easydist/autoflow/__init__.py sha256=q6-IsDJ14sqp-vBiUktkbdO-GbC0gsFzlbJ_3ECEkqQ 727
easydist/autoflow/solver.py sha256=r8-Goaildd-W2gRqaQMMcGxTVuHf3HQogniHhwVZzMc 34473
easydist/jax/__init__.py sha256=VzkJ11sElnjyetRWktjMiDQ926-xlByaZkeSR7v7OJo 1820
easydist/jax/api.py sha256=y-VPPC0Qc3j6OTMbqyelme7iTRbyB4yiljHW0lIvGKI 10119
easydist/jax/bridge.py sha256=xxgxXjOfIzg7aN-ilI4twVs2nzpQysjctdtK9-VJxY8 3893
easydist/jax/device_mesh.py sha256=57Fwc4i4NPmy0vmXYcs4CsVkmWQN25lzkRN4HUtbhOo 1385
easydist/jax/sharding_interpreter.py sha256=btNT1fTNjG7q3XqElqzyroWoaV1DLJTJHGMPMckhpfs 7624
easydist/jax/utils.py sha256=mHSkuc-IdOUlDUgnwIyt47-otgDa4qUZmoxkSgnnXfI 1058
easydist/metashard/__init__.py sha256=5lwoG8zOTyOFnvmyZszB9wMwSbCeFsniwWnhUpHW5uw 859
easydist/metashard/annotation.py sha256=2CjEXGDdTJUiQQFhBRnKFGFfNaM16mo96zvKpHUAiDU 4592
easydist/metashard/combination.py sha256=E0XBgSQQRLiklcQift4X10r-fM2uXLrkJVdFROQ9NbY 11979
easydist/metashard/halo.py sha256=MzzBRWytyi9SL57q0pSZXwsz8kYeOinIBF8Q4eooVZk 1932
easydist/metashard/metair.py sha256=OMwdw4bOT08_kdn6Y4D4KSaZVj3DHzEDk00a_74RYdw 32412
easydist/metashard/metaop.py sha256=zHQRW0XTdrVVEComMiS_Jv2tqmTssGIp0BBY8oojBQk 12518
easydist/metashard/view_propagation.py sha256=P4D48Tm3eawIUJ_R7g07SXls1IigdiyfUQl8H3QU8LE 5454
easydist/platform/__init__.py sha256=Wcl6cf4c197C69rSJ6B5SNSF7Vp6lsrWSwdEl4wf-Gc 1517
easydist/platform/jax.py sha256=2Zpf4O7erUux6nMwLSmnmoK4XRb9ctFNcNmD4rLJyHI 1443
easydist/platform/torch.py sha256=ZhdzYm09ciRRZBSZxJbCG1Kr79vhvG4Gz-fUgg3PfGI 1130
easydist/platform/tvm.py sha256=jQyQYceOaiFt-QOkin2by4_LdxdnnXJaxSAwySR0UXo 1813
easydist/torch/__init__.py sha256=xUCkZpIKNciG9Pw7twU4S4AEtTw87TWiAtIgzFbA4bM 1535
easydist/torch/api.py sha256=JUX2HeDZ8uvmlCYo1Io_OfRRKsC0ETq-7l7Ij9Z0SI0 7518
easydist/torch/bridge.py sha256=i12M8WBItTNRfIkpzjLg9Z_QFzkcGnCEygQtCFkrFG4 11211
easydist/torch/device_mesh.py sha256=0Q6tPH4uCsyDwCDXMdlI4_NRg8AMKr65FofVJ1jyQeY 2467
easydist/torch/mem_anaylize.py sha256=thibT9p4U03IjCdlnb1OC8x1-kt_wL8AvzbEeSE1ej4 7543
easydist/torch/preset_propagation.py sha256=A5iRHKzr2O40yyjR0mTEmo_KluHYOVrNHxqtoOpley0 10523
easydist/torch/sharding_interpreter.py sha256=aLQQaQC_gwPndOD5kOZLxCSfZ5s7ZQ-UZzqJ94PdRzQ 13870
easydist/torch/spmd_prop_rule.py sha256=Ppvs4YOvla8IkRQHaRch9BXJF40jlEnmyxXzms6nc94 31566
easydist/torch/utils.py sha256=kjgCd1TkJY-nJI0F2OLmgifpY6i9jh57shK0EFWkQD8 3531
easydist/torch/experimental/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
easydist/torch/experimental/api.py sha256=bk3lX2pprifAJTiKrIhSOSt6XwxAs87cQL_OMm2mfCQ 7392
easydist/torch/experimental/compiler.py sha256=uQjNDBF91QTt56NB-6QAm0x9YxpMlF2UGeGwJAmyv-g 16800
easydist/torch/experimental/decomp_utils.py sha256=KnhRRihFJdreoMzI4-pWhyUZ0H66Hj4lPhcgRkFbajE 3937
easydist/torch/experimental/init_helper.py sha256=-iuWli_oO9N-aQ9yZQobLsW6-LQE8KUuEg5xrTAykAU 5357
easydist/torch/passes/__init__.py sha256=JEfDwrsGJhXD54qlSuTazana65c9OqPYv3fGCvo5VxQ 1039
easydist/torch/passes/eliminate_detach.py sha256=poQ_zA226rZFBDLgmRn1AR6rgKJAmPqZwSGIyNQS6Yo 1061
easydist/torch/passes/fix_bias.py sha256=0Too0oHhkkdkO8kz7w41sCVN9bTMVZxaSxoCLSGMAuw 2646
easydist/torch/passes/fix_embedding.py sha256=Mntj0uOefLj4OR3OFpDIo__3ZvwtyCT9xvqltUPBvj4 1544
easydist/torch/passes/fix_meta_device.py sha256=tdiLQ--XR5mmWt1kWt32x9viMbqfqgasdnQn4VHT0CY 1230
easydist/torch/passes/sharding.py sha256=FwH9GWjL6frtJS0ZtGALqxxv39FdoqOZuVJYftyaKok 7282
easydist/utils/__init__.py sha256=rCd50yttohr1sAUDaRFi6iR-qMpv2MzGrmilrHsw4a8 986
easydist/utils/testing.py sha256=XUMLHJy9jdkWwkNZK5ZpcX1VUooUYeipXCHI4_3ktvI 1565
easydist/utils/timer.py sha256=joablFVcX4RHKffOWqxVNkZmVVwsUDwWCzBtvIHWdWc 3409
pai_easydist-0.1.0.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
pai_easydist-0.1.0.dist-info/METADATA sha256=IKqgZGu7gBso7eSx7rSZTPc05HnD7Y1LotqHP2tMyc8 3632
pai_easydist-0.1.0.dist-info/NOTICE sha256=NED0OfMi7q6iq1pG5gdpO8hwhDHPM5QbDiigMrDFPaM 17330
pai_easydist-0.1.0.dist-info/WHEEL sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo 92
pai_easydist-0.1.0.dist-info/top_level.txt sha256=2mlTnwDq-BmkV6UGv6EmLcdr4ONW75kVdIWiuEDggVA 9
pai_easydist-0.1.0.dist-info/RECORD — —

top_level.txt

easydist