Wheelodex » Projects » detrain

detrain

View on PyPI — Reverse Dependencies (0)

0.2.6 detrain-0.2.6-py3-none-any.whl

Wheel Details

Project: detrain
Version: 0.2.6
Filename: detrain-0.2.6-py3-none-any.whl
Download: [link]
Size: 16527
MD5: 210412fc3e9591bf883d090759f75b4a
SHA256: 651bd12301a8011746797eef64a7a3a9c838f0d44b622f5e7aec8cdb0415a40e
Uploaded: 2024-05-12 09:44:19 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: detrain
Version: 0.2.6
Summary: A package for distributed training & model parallelism using Torch
Author: A2N Finance
Author-Email: Levi <levi[at]a2n.finance>, John <john[at]a2n.finance>
License: MIT
Keywords: torch,model parallelism,pipeline,tensor
Classifier: Programming Language :: Python :: 3
Requires-Python: >=3.8
Requires-Dist: torch
Description-Content-Type: text/x-rst
[Description omitted; length: 3082 characters]

WHEEL

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

RECORD

Path Digest Size
detrain/__init__.py sha256=UZ7kHTPoVgb1SyM5BSajCWOmIS3Uh79KNmVVvnlGiNI 27
detrain/fsdp_tp/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
detrain/fsdp_tp/evaluation.py sha256=AMz4K91NNdAd1fhIJ4_TIjeI7XOwzqsTIXqayZH0Qnk 1775
detrain/fsdp_tp/log_utils.py sha256=qNPdXmULVTTsOhRe3KYz5OXMAlRad4PedbiGc_OdoDI 643
detrain/fsdp_tp/mesh_utils.py sha256=9Xk2r3GOSSlBDAxD6JRWoZp27F0kmjz1yWGXFb5Km8Q 914
detrain/fsdp_tp/model_2d.py sha256=v3BD2GKomOaftstux05adjNAI44VZxEolGO4Ln9xGFo 795
detrain/fsdp_tp/train.py sha256=813ggRL_7WYTcNmc6lkd-L7GeRMv9_hU6R3aFzs-46I 223
detrain/fsdp_tp/train_eval.py sha256=XgLTtqXs7HUuatPs4E-Xp9F91bY3gIcnnFQl1pfEkuA 571
detrain/ppl/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
detrain/ppl/args_util.py sha256=7f0zOn2_ELqCcqOui6MhHRHWYzUUNTliKHAH8fu1gGA 1153
detrain/ppl/dataset_util.py sha256=gpxl2djKyYUf46M2kIcvmgtg6rcOFkqM99RLG9SrHU8 2933
detrain/ppl/dis_model.py sha256=zRV7SSLQm8KyGoYe67hDoANf2OkcLSz8d1L9U023CWk 2480
detrain/ppl/evaluation.py sha256=hkWfhHkOSxvk56qlumyOYc-fxSsHFXXWcw5x6a6zUw4 924
detrain/ppl/master_node.py sha256=ih3VC6-3zj8O96szHxLTNGM3FFnf51fDNEdKU4kGG_w 614
detrain/ppl/train.py sha256=vCU4xaLfJRe7jNXFXFW8c5uL9qNbMZ06xIOrTdqCW8U 1270
detrain/ppl/worker.py sha256=6oRnmcn9SBrlsIDS38FUTklcN7h8DVb9BBjX8tUYkm4 2311
detrain/tp/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
detrain/tp/evaluation.py sha256=44vvkb02TqaBRkQUnWAfMk8ZhZXIjiy4vfr3gBuP9NU 1763
detrain/tp/model_utils.py sha256=Fwny-enObp3c8rrIqbchtLOsm0drLamopjV8CE2CR6Y 1126
detrain/tp/sequence_evaluation.py sha256=hfIFtAqJ4idEGfJMiJp_0_tQ7ZxTH093BeyGJdtNYZg 1714
detrain/tp/sequence_train.py sha256=gKhBmiMhmjMLqTPYbeJYSs7pqvr4O299lgRcDVZKxws 765
detrain/tp/sequence_train_eval.py sha256=ZbplkGaD3kcAC40MLg3dsHOCcoui89c2GcvWNn4TKn8 584
detrain/tp/train.py sha256=hXhXBRL8DpmBOsG9t3nXAynJ9YRxA-BCrMyUxauLdqU 1122
detrain/tp/train_eval.py sha256=84ZJsTI5AKhqP25AfbpO-ici6UZiRgcZ10YgXexz9Hw 600
detrain-0.2.6.dist-info/METADATA sha256=hzNcop06V51FJ7-DOR2kgnAdutn9Tlv3fKnWGgmaJiM 3485
detrain-0.2.6.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
detrain-0.2.6.dist-info/top_level.txt sha256=DCIOOEhFpcONt0t1hT6ZV0m5KJFOY_zugIq405ii3FU 8
detrain-0.2.6.dist-info/RECORD — —

top_level.txt

detrain