Wheelodex » Projects » atorch

atorch

View on PyPI — Reverse Dependencies (1)

0.1.8 atorch-0.1.8-py3-none-any.whl

Wheel Details

Project: atorch
Version: 0.1.8
Filename: atorch-0.1.8-py3-none-any.whl
Download: [link]
Size: 578430
MD5: d278c4bf368958b12c4d598f85b99998
SHA256: b1d3cab1c84b92881617d9d03e1ba36612daf374c6e22cac19be7bdc2c61b394
Uploaded: 2024-04-12 06:32:23 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: atorch
Version: 0.1.8
Summary: A pytorch extension for efficient deep learning.
Author: Ant Group
Home-Page: https://github.com/intelligent-machine-learning/dlrover/tree/master/atorch
Requires-Python: >=3.8
Requires-Dist: fairscale (==0.4.1)
Requires-Dist: apex
Requires-Dist: flash-attn
Requires-Dist: deepspeed (==0.10.0)
Requires-Dist: transformers (==4.31.0)
Requires-Dist: networkx
Requires-Dist: pyomo
Requires-Dist: pynvml (==11.4.1)
Requires-Dist: grpcio (==1.34.1)
Requires-Dist: grpcio-tools (==1.34.1)
Requires-Dist: fsspec (==2023.10.0)
Requires-Dist: pyarrow (==12.0.0)
Requires-Dist: pandas (==2.0.1)
Requires-Dist: tensorboard (==2.11.0)
Requires-Dist: dlrover[torch] (==0.4.0)
Requires-Dist: protobuf (==3.20.3)
Requires-Dist: safetensors
Requires-Dist: GPy
Requires-Dist: pymoo (==0.5.0)
[Description omitted; length: 347 characters]

WHEEL

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

RECORD

Path Digest Size
atorch/__init__.py sha256=Vy5IZwDHwIKC9CBE4oqr6Y9QF3yOrpV5xd8aabICPGs 958
atorch/amp/__init__.py sha256=faaBAtNc1_5EiBtVuZ5xarKpxOBBtVc8abHJH_RjdMk 255
atorch/amp/amp.py sha256=APe-aym_Hky6CipREVSppAExje55gAfhpzvOGo07OkM 3412
atorch/amp/hook.py sha256=2r1RXbKHbh5QRVT9U-q25QV2zJMbIoq7YbiZk2Gy60I 868
atorch/amp/pipe_amp.py sha256=ky_0Yi4jKLHo97CZw4UpPcwydziG6jq1kUXoOcM2SuY 12488
atorch/auto/__init__.py sha256=5GppCkqhC0MXbVndBHGKCk0XhHr6vFvBXwsA1EcBylM 83
atorch/auto/accelerate.py sha256=n6AWTUs6LbDOZE9HjC1evQ7S6PGbFIKTYF0yUpIyDj4 28847
atorch/auto/auto_accelerate_context.py sha256=vGul6an9wE6lAh2WXx9aamVjpSmC-1bcwL-XWE5JkPs 1095
atorch/auto/clip_grad_norm.py sha256=sPw4Jv0DTdRIZiI1oWe13usjxmXl5_6aa1wtkiTVjX8 13550
atorch/auto/device_context.py sha256=HOiizEAEhHFZIHvJK0JPCasyNmutanmd_nLCL9CJhuw 6456
atorch/auto/engine_client.py sha256=2L3rBUDPLFGZJx4nYjm2Sx3t-Dkzu-q4pO8DpYbAeGo 2480
atorch/auto/model_context.py sha256=J7qzhy4zXq_ITxQ7Ejvbhl6eZZofOO7-_lym8f3anso 37647
atorch/auto/strategy.py sha256=BTDVlv_hNgZX0rZ0-_ouZkGVpV6VG0AWAAI7qVEvUQo 4803
atorch/auto/task.py sha256=tTf8f8FJB4klUgKGbAFM7mEz3FuvZ-Mgg_bjy3TdNbI 468
atorch/auto/analyser/__init__.py sha256=4C4jMmmfnkX-s4eyoK_SuOUpTwNwABJ__9wHjaNips8 31
atorch/auto/analyser/analyser.py sha256=wtJryeGZqLe17UyUeBeuDf8Cu4OAUhyfVTCSOWNkmw0 12008
atorch/auto/dry_runner/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/auto/dry_runner/dry_runner.py sha256=m4Pq0sq9p0YzimoVv_4V8zCCOGsS0N9A4ulCnneQieM 6435
atorch/auto/engine/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/auto/engine/acceleration_engine.py sha256=oEXE4YI5E4yOx_JmmjZrLodKQ3e0oQim-deTTpKf0gY 3241
atorch/auto/engine/analyser_result.py sha256=DymBB8uPltZax-ymGQJ0pP2-AgrkV3JsGJbrVge8I_8 948
atorch/auto/engine/client.py sha256=vR0CKK0hjJoqGi7KYFLIQVaqMW9eW_O1uy549V6rkzc 3064
atorch/auto/engine/executor.py sha256=x6EQYK0xKESJIxvGvxH8j-Dkl9nVRFnbCGOf1OEDX-A 11304
atorch/auto/engine/optimization_method.py sha256=o78lmoXSdJbAHIaqX1oJKtKjCAnLri5J7_pJcDb6LUE 6810
atorch/auto/engine/planner.py sha256=ddNn1bYvtmH09DX7qllk2cclq9rl-4q36mSfSmo_fhM 4587
atorch/auto/engine/servicer.py sha256=Bfl1EuOF_66tsHmQVWRm3VbtDDbRiZyT8HsD2Or3ISQ 3035
atorch/auto/engine/strategy.py sha256=TZmr4CKBNTBm_aQ9TOw9HpjMGm4ZQnMgBK2ZziOpEio 6306
atorch/auto/engine/task.py sha256=qxENGSacFgam6b5J-7COpY54gYDdzwE0kjd5bf67bl8 1693
atorch/auto/engine/sg_algo/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/auto/engine/sg_algo/bayes_opt_sg.py sha256=u3xtRDEEtqBiYrWKHBmxzeyYGiYDB-idGjDpIlbzdTY 5713
atorch/auto/engine/sg_algo/combination_sg.py sha256=DLK_iTlLs6JYwd80O3DKzVFuFCySZ2GrPxVByxSdLtI 2271
atorch/auto/engine/sg_algo/sg_algo_lib.py sha256=9EK2ehaQ2mhRbSVaXcNu19sQ9LvqdfDWI1RnJe9JcUE 896
atorch/auto/engine/sg_algo/sg_algorithm.py sha256=HPHXpfcPRWKUGFCim-qa9if6-bomX3DRXPyu1Nh-xPY 886
atorch/auto/engine/sg_algo/utils.py sha256=BiKB0-CkdYyZkXUbaJ4He0iWVdBHBhy5Zxvd64WeJzE 14213
atorch/auto/engine/sg_algo/hebo/__init__.py sha256=5hyX6igwLMlODja5nTG-my2ksGoRbeBkZHnP6vz73tM 150
atorch/auto/engine/sg_algo/hebo/acq_optimizers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/auto/engine/sg_algo/hebo/acq_optimizers/evolution_optimizer.py sha256=8fkAw7SWmGSw2n_eZ08Vb3_tZSmDy4FY2si9gYlanMM 5851
atorch/auto/engine/sg_algo/hebo/acquisitions/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/auto/engine/sg_algo/hebo/acquisitions/acq.py sha256=7rGD9KywH3OsfbNgqrxrWNM8HIJbqMvHyyiRA4mLZwg 3395
atorch/auto/engine/sg_algo/hebo/design_space/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/auto/engine/sg_algo/hebo/design_space/categorical_param.py sha256=bg9xBFupbfmu6qiLpbtbt3NIo4xtCs3OfVVZAWi76-k 1434
atorch/auto/engine/sg_algo/hebo/design_space/design_space.py sha256=A51xXrpSyyTwF7zIDoOTIgiCFwCZDM5aBPlS7fTmO8g 3119
atorch/auto/engine/sg_algo/hebo/design_space/param.py sha256=Tmero5myaxZ8C_6LbTuAQkv2h5D8FnRxrDHUhAbXsCg 1063
atorch/auto/engine/sg_algo/hebo/models/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/auto/engine/sg_algo/hebo/models/base_model.py sha256=nRSzQh2mG90U0o_uQOnTIhg0P3e9rcQHO1kyZyFFpDM 2105
atorch/auto/engine/sg_algo/hebo/models/layers.py sha256=pDEEkXxvIr8y5vt6WgvxNu_I4uO66UszWAifV1jJX4c 666
atorch/auto/engine/sg_algo/hebo/models/model_factory.py sha256=dPBjwQ8UDempy5wIeO7d_4RZ_x-6VpFe2yg7xskIDtg 598
atorch/auto/engine/sg_algo/hebo/models/util.py sha256=YMO3xfVAQcpceXpv_cU1U_mRHwBM2Hk0zeHCYfnEslI 541
atorch/auto/engine/sg_algo/hebo/models/gauss_process/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/auto/engine/sg_algo/hebo/models/gauss_process/gpy_wgp.py sha256=sOVf2C3jamODE50q060gvmpOGteWQoxRrMsgDTejbec 4516
atorch/auto/engine/sg_algo/hebo/models/random_forest/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/auto/engine/sg_algo/hebo/models/random_forest/rf.py sha256=RER1YKjVeV5Eur0KGe4gaRKiVbyowdldvYPqh-5AXAA 1815
atorch/auto/engine/sg_algo/hebo/optimizers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/auto/engine/sg_algo/hebo/optimizers/abstract_optimizer.py sha256=lbUTNd8-s4zbjo4bV4AqEUVjix5cmmZVoI0mVVvHThA 1174
atorch/auto/engine/sg_algo/hebo/optimizers/hebo.py sha256=PZBnZ3PF1r67LfB8w_yTPIH3tia9qoBZOMk2inzrMqg 8272
atorch/auto/engine/sg_algo/hebo/optimizers/util.py sha256=anWTTbI7MGck8y7fv7b_mhcQ8pILHKp8CqpzL2CmWRY 1608
atorch/auto/opt_lib/__init__.py sha256=hm1pfI_MWOYQ8wjbeVwxU37TORGAhC6NHaEvVfBK6oE 54
atorch/auto/opt_lib/amp_optimization.py sha256=GIDWzeOTmmgIgIALTZKcSqwhIiPungNCgVJo6USAoOY 16692
atorch/auto/opt_lib/checkpoint_optimization.py sha256=tVAkRpxh0f7RZxZ-7-_RdRvlerM9jZYrFyfxfSTIULI 10362
atorch/auto/opt_lib/ds_3d_parallel_optimization.py sha256=kt6LltZWUGewT00MlwrFuKZN07dO0dKeZf-r6pyCpSE 7195
atorch/auto/opt_lib/dynamo_backends.py sha256=nASqKJuDiGzaHH1rsYrHA8TJv7oUA-8ozKb-nSPRJLs 356
atorch/auto/opt_lib/dynamo_optimization.py sha256=i4G91wNn-VzTSb27uVqe1vhftkvTTxvgE43AHHf9eyA 2639
atorch/auto/opt_lib/half_optimization.py sha256=O18VK70CTD03olzJm568DDaECG7i1shDHHjL_FVxrZ4 1856
atorch/auto/opt_lib/mixed_parallel_optimization.py sha256=T7EsDOzMX86ZuPb7GtDDwz6Wc-BGHFPeb0fZq_IrDTM 13786
atorch/auto/opt_lib/module_replace_optimization.py sha256=MHGTFm2UsUojaAz6LIHqEMSTDXehvwGjmHWun_PAxlY 7217
atorch/auto/opt_lib/optimization.py sha256=OKO5uf_W8mvj0xzeRLJws4PxeTm07Te06N9z1n7-Qp8 10892
atorch/auto/opt_lib/optimization_library.py sha256=6khls3p-kyzb4KcMHASVVri9zf9BB0iffhHZAYoOoZg 2787
atorch/auto/opt_lib/parallel_mode_optimization.py sha256=f-UylUODH1tpEXQvZNB55N773nIjRfMN8XnxoyrjJHo 2112
atorch/auto/opt_lib/pipeline_parallel_optimization.py sha256=062_oAxSiCv9vXE6UEHfV90wb2KCeo4QWWs19EJhgjs 11721
atorch/auto/opt_lib/selective_offloading_checkpoint.py sha256=6hskQusHkWk1WaehMQUb-QUT2sRlOEc6nvRyrw0qaQs 10728
atorch/auto/opt_lib/tensor_parallel_optimization.py sha256=uD4hkk8zzTuBTrJoM0IXVqqExIPrKfUfTQ1OHEDeGI8 7992
atorch/auto/opt_lib/utils.py sha256=7Ghc12Oduc4jTkl_LSMDIdUsgBCu2av9CYr_t0cN9ZU 6182
atorch/auto/opt_lib/zero_optimization.py sha256=jcvxpIRI--SzSeM87-UyIczgRK21PLfk14Q9AQM5k90 22731
atorch/auto/opt_lib/shard_planners/__init__.py sha256=TbUvdmFvOAKiwJrh36y9G6tkW9N8TWOVyWt3vHnbxAg 241
atorch/auto/opt_lib/shard_planners/base_stage_planner.py sha256=EhgthFiK1PuCu24mJJ9TIp90Q6GjYgSkAFcdpZm676Q 5145
atorch/auto/opt_lib/shard_planners/base_tp_planner.py sha256=TX6Z-R3C_nI_OHF5R1KDo5ooeVWNMX7oL3PzyF4pHnE 9408
atorch/auto/opt_lib/shard_planners/dim_planner.py sha256=ORKrzIWX0NsRgbpuHS5rEVDfYhFcDGxjcqMayCVW_eU 8974
atorch/auto/opt_lib/shard_planners/mip_tp_planner.py sha256=tv53ovNNDQF5dm6yAm5Y9cxeWzGi9r4KnFwSAiDBQ4M 22977
atorch/auto/opt_lib/shard_planners/topology.py sha256=cUR8kaHlZu00CjEmQ3UYf1Qkuj_sRU2eFwT3WSW53WM 3471
atorch/auto/opt_lib/shard_planners/utils.py sha256=msJUuzpGHqQZxjNbNLh3zFuGWGaHhK2oYqKNdlf7b4c 26926
atorch/common/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/common/constants.py sha256=1vsxKPDJ4QDw9jEvtBQDSY39cM2PwTpKNuWVc_UjISI 3383
atorch/common/log_utils.py sha256=GKPY2F5HX58PHH3h8ytZPTELN9gc_HUTFsAt_m6ZixU 3340
atorch/common/util_func.py sha256=6Qea8NDi-mLOc3Qbd-ysVlGUqeBvsvRxf2mP4KS2F0s 6077
atorch/common/file/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/common/file/file_io.py sha256=9BgUxdh7o9KEGnuFguhLkIt-lsfJxdRmJNs6BPJOZ-g 845
atorch/common/file/file_system/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/common/file/file_system/file_system.py sha256=DPbmj-JhiwS83YBo1pVIuoymzqf_VPW06134Scdwmxg 4991
atorch/common/file/file_system/pangu/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/common/file/file_system/pangu/fsspec_instance.py sha256=VzfKkeMjMZLSRjRdU_dkNqlGXtMXbkcGY3DP7SMHukY 13710
atorch/common/file/file_system/pangu/pangu_file_system.py sha256=EgPDOkdpoOBBI71MA4qXWXm3AR2A_DOpUzU98RDKT8U 17043
atorch/data/__init__.py sha256=mHv0gvkEvucRaZmC8KmzTabD4Tz-OIScrEgAr57hqG4 633
atorch/data/coworker_dataset.py sha256=b4iPlNloD0IAXEmIF7IJ5Y5hCly2HeVokeGcgNMy-R8 17310
atorch/data/data_utils.py sha256=yBhMkvUusMKQ29evN1bM1_EXUj0B2sxcgLmD8pZpUfc 1347
atorch/data/elastic_dataloader.py sha256=mrSDrWTLxehdCRnzhqokNGBB794zAKjZUQiQ-Dd9Qzk 13694
atorch/data/elastic_dataset.py sha256=xOU70QiUwvtrExfgoKF4dXwcofEtNg4IiolOO9xrTNM 4771
atorch/data/preloader.py sha256=hkYVHJ22dE7bPV9TNOO6ihmDWVqPkPtRw3_mi87rUXQ 6113
atorch/data/shm_context.py sha256=iUwp3xcoXknGf-CLizR0qkVFy4xcPx3SqdDFHUFx3HE 27904
atorch/data/shm_dataloader.py sha256=2kNeX83Xcy2_RH-WRz5THQ-Q_VPR-gLA4xo_aJ__6Jw 10185
atorch/data/unordered_dataloader.py sha256=fMRrCPK5tzEPEFsVl7glxKjav6k5rKtRp1t4HYFfwy4 3396
atorch/data/unshuffled_batch_dataloader.py sha256=-mYLnymYcnoWdWovRNkfSW_96yK6GgTdz8285tenzsc 1425
atorch/data_parallel/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/data_parallel/adp.py sha256=-lw6ABlDyPiRePMZ_7s6dG09rNW6ewGImQaIaIk4bGU 25559
atorch/data_parallel/auto_wrap.py sha256=720Ewtqw07xI4g1CxRsnof0WmyOydXPfeDqqXpG-_Sg 13332
atorch/data_parallel/wrapper.py sha256=oarEzeRLlcMWpWidiZaiOifZINUqWwEIeSQabxNKcx4 3562
atorch/data_parallel/zero_ddp_mix_112.py sha256=-THdMHvXANJgGq0EeVBbf1_wQdzVfuScx4VCeQvTboo 25890
atorch/distributed/__init__.py sha256=VzPm1UfPpoMP6_XMZHNK-PblQcOg6egxEacU8BgISf8 261
atorch/distributed/distributed.py sha256=A7N7yJX62YOlkmYhZ6DVzCUSfKdmskXYtJJ0V-G2gvo 28079
atorch/distributed/elastic_controller.py sha256=qyy8BHCOIhtHIje-a6d_O-BkRYL31M251WT1DVqB9NQ 465
atorch/distributed/elastic_trainer.py sha256=V8KJdKdHuiEe_BRoMEdbVXE38cCpO0j9aMRGr9dqujY 3405
atorch/distributed/hooks.py sha256=7KQNyYtDU6XrekztrKQQz31S-JagyPjZ8OA-6BVNWNA 1253
atorch/distributed/launch.py sha256=sAWY0P5AMxOz2cQSg8Niuhe1tY3BtVSjw4CCPaISClw 19071
atorch/distributed/run.py sha256=llXERdpqNg2YBdp6tDMdnHNKmaEF9Z1Nq3ATnb_iWIo 12424
atorch/fault_tolerance/__init__.py sha256=4upTsfblcbQZxgCnUYEpvA3ALhjPqUkkGEcqkzNnjpg 68
atorch/fault_tolerance/api.py sha256=wLXxCJRm8dkPEAjZbarGh2pldfOxdnpMvqOwr2SlBes 4825
atorch/fault_tolerance/custom_agent.py sha256=x_PpQGG4Hvopic2W9y78rtrBmsb8MI86F1mKln27XQI 8293
atorch/fault_tolerance/hanging_detector.py sha256=tYFGtHKAgZN-BBRG09QbsUCLEBndDK5UuludpFYpecw 5659
atorch/modules/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/modules/distributed_modules/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/modules/distributed_modules/activation_checkpointing.py sha256=DtKP5ByzmZ4Qv5Lf0XUo-eeVElVgtodR6Sk-TE0Lt7U 14763
atorch/modules/distributed_modules/cross_entropy.py sha256=2vO4JXfsaNly-car7z7oAVQwkjlHBQ7u5maz0ohtS5I 5688
atorch/modules/distributed_modules/layers.py sha256=Gmy1QAASienm74iSrKfzK91EFCZjhKwfmdtkuVCfER4 29411
atorch/modules/distributed_modules/mappings.py sha256=9BQH-Sf5ZaXWQ8_Vg6B3Oe-7O2czC1KCCG42TW1tIE0 16790
atorch/modules/distributed_modules/mappings_registry.py sha256=j2a56W0lOQxwchJVM3n6gMPZgDkFym2NGP92UCsP8OA 7633
atorch/modules/distributed_modules/materialize_modules.py sha256=icJiWK_5x70_htHUewo8cWob2pi9XGzB4avxrcjWoU0 1269
atorch/modules/distributed_modules/modules_registry.py sha256=j6s8PpU4bSVHLuTy1xVlm9FCdhREsV09FGGPV-P6bkk 54608
atorch/modules/distributed_modules/randomizer.py sha256=Ja_GT8_Gnn9IqlTLwlKQI8oGyjhfscSTMG1RIkI66uo 5436
atorch/modules/distributed_modules/transformer.py sha256=zJugPudfwLj_KOutOBBGgwr8Ty8VHjx46yERp2BE85g 75800
atorch/modules/distributed_modules/utils.py sha256=bKulFgK3f90Y6cKIzlpCRHdMDPTO9ShI_S-j4yKTe14 7551
atorch/modules/distributed_modules/compilers/__init__.py sha256=9xHmv75S0Ic39I6zDFD1HtouJS3uBoqY_rctM7TO3Ow 147
atorch/modules/distributed_modules/compilers/pipe_compiler/PipelineStage.py sha256=uQfaOBQiyVU0g_hw8IyqPrELGioMm48BGrVyzVKYLPI 36537
atorch/modules/distributed_modules/compilers/pipe_compiler/StageInterleaver.py sha256=sQqS81L0w7LDHjWHLSQWxNqxkvepEs7snnoIpNGPA4Y 3976
atorch/modules/distributed_modules/compilers/pipe_compiler/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/modules/distributed_modules/compilers/pipe_compiler/distributed_pippy_compiler.py sha256=vmcIvQAcwT25uuHH59ifAWv67hJaY2nZZSIpFKw5nk8 20974
atorch/modules/distributed_modules/compilers/pipe_compiler/utils.py sha256=63Ixlx4cXyAddfXEj7ylsHVeEv_qXaLUwxD9cRZ6l58 25818
atorch/modules/distributed_modules/compilers/tp_compiler/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/modules/distributed_modules/compilers/tp_compiler/dtensor_compiler.py sha256=wIw4OrXfdHMKDHizVkXOM8nRUycdWIWptehJIniihNI 222
atorch/modules/distributed_modules/compilers/tp_compiler/tp_compiler.py sha256=M5uymcUECFf-fkd1-BYjmLyY1bQpOgFQW8QVmP37gsI 7707
atorch/modules/distributed_transformer/__init__.py sha256=LVuyK4aCir9zRsde-RvMFRNUSb-zTeGN5fcS36nt460 80
atorch/modules/distributed_transformer/commu_utils.py sha256=ljzY1lpNjdEzwfCg3GbO6x4IyCJejlawXAKwWXOh4hI 3173
atorch/modules/distributed_transformer/distributed_attention.py sha256=LCMkkcsmvRCdPhnmWJGSAJ6t8rmGQO6JPcmbA3ghl8E 14623
atorch/modules/moe/__init__.py sha256=ddTRkRFLvmEws66--QiOx-dKT4mdKd_8-Q3TxBjjwBM 295
atorch/modules/moe/ddp.py sha256=xhYn6xEzkzkoEY1g8hS8zYNj_U-OkAb9orx5xPzSgvA 16005
atorch/modules/moe/inject.py sha256=njrdSMjILOL1WJZwZFJxROaHNoSiuF7R4tlU1yQyHfM 4291
atorch/modules/moe/moe_layer.py sha256=PQ4eIZx81JcUO-D-UbLnIpQAmpSJuzpje4FhFUhUJvA 25005
atorch/modules/moe/switch_gating.py sha256=1OfKZFJ9YrfvQKJCUUnoqiHAblM1sAavRka8uzzRmkk 6683
atorch/modules/moe/topk_gating.py sha256=BTScyg2dkt3WcX5w_ojzTBgvaDbsiT7mSKr7tKw0B4o 5799
atorch/modules/transformer/__init__.py sha256=8jGPBsMO4Yc-rsAO2CfZ_dn9kIu_8faBo4b3lTE7rm4 37
atorch/modules/transformer/_fa_api_compat_patch sha256=iOfGTPpaTarJ4hVaS6VoQKK7iDpn7N2LUwgAhlASn78 5210
atorch/modules/transformer/cross_entropy.py sha256=Xv32RhMM53jCLrhu12MEEShzwY9-9YpzJwElZoqAqZA 13580
atorch/modules/transformer/inject.py sha256=0AVocffvENX0ppvXV8EarYhvmRw8yQiUKtAeK2qVSfE 8248
atorch/modules/transformer/layers.py sha256=ua6S8q-rT8EALWypxOJITAod4zmO4l07pLl-MgWHJ3o 67467
atorch/modules/transformer/linear.py sha256=p2bu3WT-dyvwWxdIYhpFlTxuFeV-cqQrUgs-y7zEk0o 2682
atorch/modules/transformer/losses.py sha256=sbMsYmzqZFArbHFIGh-1EOhzjXEBZJQTvk7q7Yh2cow 374
atorch/mup/__init__.py sha256=T5q60GRUstHUYwNSsoIsGsDOVcs7C5UlDLborv5CpTE 244
atorch/mup/infshape.py sha256=QOolFbJkTInt-Mr77RzrRHOIjFx6sCJLE5VBT9tTBUQ 4232
atorch/mup/init.py sha256=QZKlSX8YIWNyHc_3KmjyahAfbAA27gSk-nZbM0_w0tk 8553
atorch/mup/module.py sha256=5vLI9NGAVO-S3D8-ZI54W2svq70sYwRN_w7Aq5_WAhM 11021
atorch/mup/optim.py sha256=X4SdpBKh7588kLWEnEXM7DXVNVl3-14tFY-4CMTNDbw 5867
atorch/mup/shape.py sha256=V-L8eCgD96FAS1idKmjwWDnlK3SGk859pNBouGWdTM8 7971
atorch/normalization/__init__.py sha256=jS_nSmlS1t03tY1Kvt84e-XU6aURJf2xMqWOAnsWorI 455
atorch/normalization/layernorm.py sha256=YlQ9TP3yN5289T6CauqgkmxPQqq7YkmJxpVn9JiEF9I 9000
atorch/npu/__init__.py sha256=j9kJhxhFccrmIpJYN_5rUUWawVArR0KLtcVWZ7qNOVA 2681
atorch/npu/layers.py sha256=H-d9jQlA67nbh0XpzFUMH_Nm0CjNwFc1nv8jknPDyhc 6090
atorch/npu/optim.py sha256=1lbnfV1xiEb55091M9Hyhi3SqISs1KbtNr-gGjdMtZg 7676
atorch/ops/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/ops/git_version_info.py sha256=0UWHC_5zh71LNENaFWCacCsC4PsbHi3gKOihQBLmzeM 950
atorch/ops/git_version_info_installed.py sha256=M5V245OTobrxnMBjsmJsFieeKSSlD_JPq6DVv-jWztM 347
atorch/ops/accelerator/__init__.py sha256=gWQg_tMCqUv4mjGYxXV-C314BDS4IEdLEV4rGPwfXUI 271
atorch/ops/accelerator/abstract_accelerator.py sha256=Bz7eo_CCRCr_V3eO_8xw7aL1WqVftUU-M5cd5e6qzHQ 4732
atorch/ops/accelerator/cuda_accelerator.py sha256=LFvghhqUrL-Z9rRnWZXfouPy4ZNNjB1uyrLbaRBc7b8 8910
atorch/ops/accelerator/real_accelerator.py sha256=jWxYWBNNWbDpL6Vhk_4_ySLCV3rHyT-GdcylseRMvZU 2478
atorch/ops/csrc/includes/conversion_utils.h sha256=Rj9QBKdYTLrkCovxMI03IKQKb153SrtgE9WYJS_PnLs 12017
atorch/ops/csrc/includes/dequantization_utils.h sha256=q9gZ4uSR9Kq1nU9bFu7oFMkWUV9kflzX78ZiyUQdXuo 7001
atorch/ops/csrc/includes/kernel_utils.h sha256=iZ1AAiQL8ipcNdSEN4itVH8xPIXiAI4XWrnpz6lGfVg 1202
atorch/ops/csrc/includes/memory_access_utils.h sha256=d9QZPa5WK2-5WRyG-4bUaRRX2KTeIHwFdGEQ0zD1774 31857
atorch/ops/csrc/includes/quantization.h sha256=n9hDanF4DDyD-80-VHX5kCo-WcqMr-4K1ixsd5FmhpI 1838
atorch/ops/csrc/includes/quantization_optimizer.h sha256=N3rFhlmkq7f1CPoWwA17c4du-sSvrCNxJMSSQv_5ArY 1884
atorch/ops/csrc/includes/quantization_utils.h sha256=jN8Q5FvenppiTMXNSk0TQqZ96ongbgL1xjiV5L8TDWI 16931
atorch/ops/csrc/includes/quantizer.h sha256=iGVZSHcz3mGjOc8rxAdaiXRxA46PLKjeWyIOFv9uoMQ 411
atorch/ops/csrc/includes/reduction_utils.h sha256=bflMkEVgaaJ2Ad1DIDQYmJo6qjvUeVpUJ5_52S7TLQc 17562
atorch/ops/csrc/quantization/dequantize.cu sha256=10XY8HB2YC_vtj97HQS2pECyWDUgzN45JCj6urqIFMs 3048
atorch/ops/csrc/quantization/pt_binding.cpp sha256=0oCSOrq4ijzezE7cRL0EQzYB8LuCUvZFlOPAIZ5_r58 8061
atorch/ops/csrc/quantization/quant_reduce.cu sha256=IVmRScHksi_8cSktVuP2x1jitF-II-hy58TXn7DNIog 9192
atorch/ops/csrc/quantization/quantization_optimizer.cc sha256=Uq0fW5t9WKP9EGt6uSEgvcfuIOtCJid5cYaV_XuLpgY 2791
atorch/ops/csrc/quantization/quantization_optimizer.cu sha256=HXlUaHHdqtQ6xiihVDLW2T_FeM_IYVUm7VBa96aw0aI 24575
atorch/ops/csrc/quantization/quantize.cu sha256=jUYr2QBDpf7AQyn1Qj9td3g3EP7eIi4DmwJNFt_Jj8c 6418
atorch/ops/csrc/quantization/swizzled_quantize.cu sha256=RhE2ZmaxOzMbDf7dbjYrDHrxBycs37IACjJru8q9LDI 7576
atorch/ops/op_builder/__init__.py sha256=AnKFf7zedD54AIsUxnXWzh8Mp0J5P1CXB5O6uVk0tgg 2091
atorch/ops/op_builder/all_ops.py sha256=rSIW26lYrpM7etZtNZQ80yUaTUT_fWJ06fZOkhsjDzY 1271
atorch/ops/op_builder/builder.py sha256=zUxTzqhqS-GFJ9J2JAbrnLw4uxz1Er-prerxz0bS9Ss 25650
atorch/ops/op_builder/quantization_optimizer.py sha256=yMsFTJwyIQZXX0qnwSqXVVueU-woBJDCwipb29FM2GY 846
atorch/ops/op_builder/quantizer.py sha256=3C1H0M3cW0fCJ3iIkU-gYaTtwYKm5_6eNOQHIhtgGtk 920
atorch/ops/quantizer/__init__.py sha256=UOtZpuW_E_41eDw1HBY86KGv5l6AokgGLQxqvjKxIDA 2401
atorch/optimizers/__init__.py sha256=QNcw-KPs3WSs7nnQjzIc7cpL5aYuCqIWilUAyaa9NuU 93
atorch/optimizers/adam_offload.py sha256=csE2HMqiuRFWMctUeA3B23veafgTd4CUtMeSK49ZMBE 13353
atorch/optimizers/agd.py sha256=c58H0JnQ-I762xjbWzfcY3ueSrFnpPR3w6pBXkb6zU8 6808
atorch/optimizers/bf16_optimizer.py sha256=wHqG3m28gQDBEXlt2cGClkIlAumr7AgMf-4yMnz8Jew 11915
atorch/optimizers/utils.py sha256=rdE-kZQEb-Q8xyoiAu4aRHZ-C84mKrPayjz8Txm_MKY 625
atorch/optimizers/wsam.py sha256=IErIOvzd6WrwU48jk0sPxh57RwRsxmol9m6lUwPwb6Y 5375
atorch/optimizers/low_bit/__init__.py sha256=F6vweKmAWojlNar3qESYgnxFzbDXsBL3IW6w2Z6ZZ5g 55
atorch/optimizers/low_bit/config.py sha256=dcTwltNcqPelFiS-NXc15OD_7wRattk436SexpeUcLs 626
atorch/optimizers/low_bit/functional.py sha256=6YZzeU62YUmF6BiixqpJRMSnbpxgIqm71OLVNP6FR6o 19476
atorch/optimizers/low_bit/optim/__init__.py sha256=EDq0eIKpAk2IYcgg8XiZhBTJ2J-na7XAXZ2QR0ralH8 118
atorch/optimizers/low_bit/optim/q_adafactor.py sha256=Sd_SbyZ4LGBCr5Itm1KzcFinJxpz0Hnenx-DrEHCw9Y 10365
atorch/optimizers/low_bit/optim/q_adamw.py sha256=8IXD8fFu0H8VqrsPCFQorLH_AVfQ5R_Vs0h5F7qz5iI 7123
atorch/optimizers/low_bit/optim/q_agd.py sha256=xsVBCQ4HgkekOwbjoX_KdiXfFVuxcGODZoYVBHgMniQ 9927
atorch/optimizers/low_bit/optim/q_came.py sha256=SToF6Bws8wTa1qS7-5Fwnqzx5BT68nbJ6kdwqg-5sYs 10023
atorch/optimizers/low_bit/optim/q_optimizer.py sha256=tFOQbRfjgN3sb7xJvHp5RgdqU_Dj7J6ZqS9B3rb5g4Y 7034
atorch/protos/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/protos/acceleration_pb2.py sha256=njaQFtIRpb-rnHO6FzqBtCuUiinyUQh550Y7F4RQKDs 19348
atorch/protos/acceleration_pb2_grpc.py sha256=zuznfu_tehpKMLI35pyfB1XqMHMH6czvPRjPBgEN2dw 4574
atorch/protos/coworker_pb2.py sha256=E-RcveDUbrH0N-vJoLFiuLi8e_Tj4RC8o_92bhIvCS8 5993
atorch/protos/coworker_pb2_grpc.py sha256=X29YrfA028EACXZnjTTxZBqugKc_CoozIJC3k91HHqU 6714
atorch/rl/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/rl/config.py sha256=ctu2t1_XuFW6--QbpCmxT86_DIoT7WIZoMQdq0Ij9IM 7661
atorch/rl/main.py sha256=SWg2e2Y5fa5yvdUv7w_72sulX4R1sO1T0hgKToVV_2c 849
atorch/rl/data/__init__.py sha256=BJhzBkR6f8S99fjMr3MhfZZL2QDHkf8Ovj3PDQlS4a8 39
atorch/rl/data/data_utils.py sha256=p43-c-pVMzHQxj3pHSgh17CR5A5aF7xpfHBHVFha5qY 6390
atorch/rl/ds_hybrid_engine/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/rl/ds_hybrid_engine/ds_hook.py sha256=UOAvGBYoMNm5VokM-tfjRIyIIx4TtSBJl8doYfODJSw 15390
atorch/rl/ds_hybrid_engine/hybrid_engine.py sha256=Sg7JBuefdX6meBzJnSZZ7kHVzXNfjR7I17JDl8l7apQ 16621
atorch/rl/ds_hybrid_engine/initialize.py sha256=2rpj12d7rjB8rmRIdvrwjzFlnZXwkv-1op1qttG57cM 7738
atorch/rl/ds_hybrid_engine/replace_policy.py sha256=D_d1czuRa9RJSsAdJhWjoJHTHUFruEQT-QFZyfV3ErA 235
atorch/rl/ds_hybrid_engine/module_inject/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/rl/ds_hybrid_engine/module_inject/utils.py sha256=zcdS_QCh7ibHDHprD0hGx8Lo6AJo37kIk24HgzXiwwg 632
atorch/rl/ds_hybrid_engine/module_inject/containers/__init__.py sha256=fu-mdfot8zaJN7rXiZVkv6X2db03BoG1Hi8t2r5Id1w 55
atorch/rl/ds_hybrid_engine/module_inject/containers/llama.py sha256=nX0cZ5-MIXjDqyhXpl3_Zqsq1wS0q6W3exfMLXqrxBA 6707
atorch/rl/inference_backend/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/rl/inference_backend/vllm_backend.py sha256=Wo-j-8B0LjdDsR3e1fZ0eh8rCoN9GANGzoLX2fH7ExM 1682
atorch/rl/model_engine/__init__.py sha256=uDupiAJd276yh4jb7ZXAgRVVQxVCfzxhk74IyNTLcq8 56
atorch/rl/model_engine/model_engine.py sha256=H4-y3fwbXUY8uRUcuAVL6vudLgm9KPbthFbrPd5E__g 21156
atorch/rl/model_engine/strategy.py sha256=b05v8BzpGd2NGLw2m0G6VeCUQDTduKBE_6OABS1hFgw 1138
atorch/rl/model_utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/rl/model_utils/llama2_utils.py sha256=H6P5MIL4amzaWGTqdSjZ0XKTj_Arstpm2Tr54SBPE6g 6372
atorch/rl/model_utils/load_init_model.py sha256=GUD4g6vuJ0X8iXWam6suyDZ1EeGYIfd8apOZeQYJh1A 6291
atorch/rl/model_utils/model_util.py sha256=1UtBnIuozzmr9w92V_N-fjzrkqqa9tzrg6rsoRuSh6E 12237
atorch/rl/model_utils/redis_util.py sha256=Rr5Ly10mx4RqKxHYvGAs3w6W4A3iNlV9YB_RT-KWYpY 2144
atorch/rl/ppo_utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/rl/ppo_utils/ppo_util.py sha256=mGa7UYHFNDQCx8S93bEymRY4hfMFPjem_hdC6OyEWb0 6842
atorch/rl/replay_buffer/__init__.py sha256=_Hh5PnQ5GBIoVBAwc78KtGsxo5gKiSPy9IOXsvT5uEY 40
atorch/rl/replay_buffer/replay_buffer.py sha256=C8VOnWn2RQO0q0RTh64NH4zifgE9g1uLVvUwKSx3eJI 1795
atorch/rl/trainer/__init__.py sha256=o1t5KLkzwRkFBB3nbg52yih82DFiYZRX93l_ekHjl-I 34
atorch/rl/trainer/ppo_trainer.py sha256=u30GMOz6PAxBFF3D5hGnSBU-ooKwyFSZ_mzC6ddrSv8 274
atorch/rl/trainer/rl_trainer.py sha256=S_ccwiN2YwwOk4wFwNd_Yrpus1PLN2RwYn45BboKxnc 3190
atorch/service/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
atorch/service/coworker_data_service.py sha256=0mkkj9vHibS6X9wQVoEk3CIwd323Vkr5WqtZG6iWYog 2208
atorch/service/data_info_service.py sha256=3mRykB5OMBnfIpi8M27peh-XmQY-vLe1Q37kRDj6ng8 1924
atorch/service/rpc_clients.py sha256=4tKfDLeWqa9zWEUGYpkfgdvmx-d7IFv8m459oFoR6eE 3238
atorch/trainer/__init__.py sha256=IeqSgO7h3mVQX6TrvFFPjlxsIkrQB939g4W_JVuRRFU 103
atorch/trainer/atorch_args.py sha256=JfxFggF07xytT7osKDn4R8tdGTusirT-kQ9mFGLafcg 10651
atorch/trainer/atorch_trainer.py sha256=7yXAhnkk4Q3wrmovIBiD1XQN2O9zQszSDvNgZIWOGeE 98297
atorch/utils/__init__.py sha256=XdOjdSUeFYAxmhjAHSUqeoLsKkcTSu6e_m3Mw20p_3U 63
atorch/utils/ds_pipe_utils.py sha256=kb5wBRNrBCnSt7xRGq3PjizjHalJbvebBLramwVvT58 8524
atorch/utils/fsdp_async_ckpt_util.py sha256=FpczDvGyqt9wRFfTgFkZ234v-atsAEs-kVCE_ISTOTo 7784
atorch/utils/fsdp_init_util.py sha256=jcJBUuEfAxNBrA2MWmW1hQr1veuVYZQRG-CxUJmCKyk 14062
atorch/utils/fsdp_save_util.py sha256=E_eWDDGA0jnRzRJ6iVDRua5SYOFDIXe-1K7O9_Fz7Wk 41878
atorch/utils/grad_scaler.py sha256=VXkj2R7wQjewbioRtcNLFnCvZa8XBJep4nGnisiO9tw 2706
atorch/utils/graph_transform_utils.py sha256=9xm6n8ekjUbvyP-HC0tkhFYF5f_r_tiznozBQRWyXrc 5644
atorch/utils/hooks.py sha256=WDCvgA_Q43_ySZLPczqSnxx9a9DTMjXeLTYGgDbSBww 1021
atorch/utils/ib_monitor.py sha256=9X6n3T7mzZ2vDRATz90E3VVmSzZTKkbSuMRGfInQpJw 6940
atorch/utils/import_util.py sha256=qMGtsKMrN2wqmm0_IAMTUooaq1ZjP9x4oalldKPr7I0 1428
atorch/utils/manual_tp_utils.py sha256=Kg8sWEZqrPciK7zt8t49OqfdZqg2bED9Tgpc8y5VdcA 10381
atorch/utils/meta_model_utils.py sha256=j1i4OkEacMF-904cv7RGuCqztMS3hgoPR2i6GoAunLk 34986
atorch/utils/meta_overrides.py sha256=nkaTKZ6siJ1b0J1FziJcn6b0hZ8f9oDDzU8Jacrkb98 11593
atorch/utils/metric_util.py sha256=aPCo5GqG7do-h_AH1PoOYZtpOJ8EYdnw7KSZRukpgcU 2267
atorch/utils/numberic_checker.py sha256=JlvqWzDjAbhMnnhElNDHaJLJ5Eoid2f5mNyms8ZCQwo 5781
atorch/utils/parse_fsdp_mapping.py sha256=5WsCSPFHMTDsJFdH6zDwv04wxoi2tflNAWrTDwOAFDc 9514
atorch/utils/parse_trace_json.py sha256=rGdUgPQ6I34SXtQGBcw25JE-d0s7tKTBnDo_MC-XJQo 10451
atorch/utils/patch_fairscale.py sha256=xnHugClneeeaW407fJvj2BS65WuSdo_JZX-vIir_H-M 3487
atorch/utils/patch_te.py sha256=mb0KUoCvcX3x0y3LqbQwXPEWGxitKOIK-etQ_UwjSD8 4875
atorch/utils/pipe_file_utils.py sha256=dEKBb-YLRNBZc2Qzp039zn8s-VM9hUVlJpBTxn3hqnA 4188
atorch/utils/prof.py sha256=FfH4w8h-JB7xHdF7zdD-m98bxcWUi5TcIQEy8Onl8cU 46954
atorch/utils/shape_prop.py sha256=J6qIbDc5TZgCFLCLdu_WTDybLW07GoIZyUerAA0kHm8 2839
atorch/utils/sharding_spec.py sha256=wLWHcIRM_OjB1H3sQZluOtUfmMFxz7VETlOYPBS9aS4 4281
atorch/utils/sparse.py sha256=ek0trMwZbdGQRo7v9g9UgAjESczZZxDDvC8X02GWCfI 2547
atorch/utils/spec_prop.py sha256=S9OEHpoXVOZQMEMaqKB6LhwsAOQEa5borPAh5F5lQoI 6342
atorch/utils/timer.py sha256=4_pZda7ZT_7vQ5cRBz4e-ZpYU9qaSivKxSZtGUbl1XM 2797
atorch/utils/tracer.py sha256=6VWBeLtW_BNyRBYVO3Q1HvbnFstg30qmdXYoguRoggo 24365
atorch/utils/trainer_utils.py sha256=gd4XcPGEDgzQ6VslrHGZBdTcI8CbBZm-x6lv7a5gd5M 2244
atorch/utils/version.py sha256=i71v6eiVb_IXQdDlWOwYWnyMok9j29nzLGh83Vjic5Y 1441
atorch-0.1.8.data/data/acceleration.proto sha256=spmYAqmckmV29X1XkmOSwajWgKMGrj7pn1nraL3TnW0 1046
atorch-0.1.8.data/data/build_proto.sh sha256=sbTOt_HHt_F-DyfgHY74m755redknL6nHeZwIUjswD4 224
atorch-0.1.8.data/data/coworker.proto sha256=hezllszC0XadStrgrgi6J9pmlZLJj2E50uTi9aqVamE 455
atorch-0.1.8.data/data/requirements.txt sha256=6EcK-CwiZRCpwxiU3srZyvlvLESn0fF_NK8Xr2aj3JQ 273
atorch-0.1.8.dist-info/METADATA sha256=qDqP74juQzP0sbkOqvxT3HKN-nqSghepThIKl1t_mIo 1155
atorch-0.1.8.dist-info/WHEEL sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A 92
atorch-0.1.8.dist-info/top_level.txt sha256=64BT5ChWxA_sdfYF3xxVLu_I8jhTEfSSOlK5WQ4zcH0 7
atorch-0.1.8.dist-info/RECORD — —

top_level.txt

atorch