neuronx_distributed/LICENSE,sha256=B7N1Dzhhg_GAema_-w9TK4rm0cod3OJMrMmbFGG5lzk,26121
neuronx_distributed/__init__.py,sha256=G5IwPB22Njdm-PNupdWec5n90JcYS8h577AfSF2NIk4,675
neuronx_distributed/_version.py,sha256=4YlgVX0c54onPfB4b-ur-dtmRUDlmuDuxbOzW_Xv11Y,177
neuronx_distributed/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
neuronx_distributed/experimental/quantization/microscaling/expert_mlps_mx.py,sha256=ELfIN0uCz31kHmx-KWyIccSE4iRkbtAs5LBLulDUjZk,9802
neuronx_distributed/experimental/quantization/microscaling/mx_torch.py,sha256=K3RZPrxbEXh-llobvLZg5d9OgXCBeOXRWG_4dJN6Rzo,10455
neuronx_distributed/experimental/quantization/microscaling/swizzle.py,sha256=DIqFOdOij3frQXCP04qvALPeB1lZSCZEfrh2PE9wBvM,1917
neuronx_distributed/kernels/__init__.py,sha256=je-s8iUO7X0TM1MRpyGZtwr4N7s7yeBv2l8oT_V4ozU,97
neuronx_distributed/kernels/flash_attn.py,sha256=nvkt9yediCJKG2hn1yhzdSNDtHqpLM5cAziIeP27_uo,7925
neuronx_distributed/kernels/kernel_utils.py,sha256=kfi9iwlTTtIXZNzEHYFQQerZ2TqvH75lz6Ue8Az6bjE,2564
neuronx_distributed/kernels/ring_attention_kernel.py,sha256=E4otw3zwTr-RcFnCVE-mi3P8SNlTJdtLaaqQvJFdQ4o,4876
neuronx_distributed/lightning/__init__.py,sha256=sK8lYszodlDzlLVwLvNVET6S40z7kJbF5SX0xOvFPic,369
neuronx_distributed/lightning/accelerator.py,sha256=vDMiu0o5xHh6tGwHT8dMwI93kmjFdv2fQWvcmG55g80,4894
neuronx_distributed/lightning/checkpoint_io.py,sha256=Z9z4ilfNMic3TECe5bUkFe2qbbmsngBAbEj4i9GRRMI,2726
neuronx_distributed/lightning/launcher.py,sha256=usp4cJrU4SpxI1K-Mq2g7QaX3wRrn2Keg0B-f8XI7ss,4029
neuronx_distributed/lightning/logger.py,sha256=p7p_KY49NpVbMAsmH_aWfljbZ8RAlsN3SFO6sFQzp0o,5804
neuronx_distributed/lightning/module.py,sha256=IvLE7GV74Z4wwN3kGWlaF7RFWhZX0VB2TKEDMI97ESY,14131
neuronx_distributed/lightning/neuron_hooks_callback.py,sha256=rSMT8WucqyLJe9dH50G6CzcdQjErU_hz3srfp6vTfAY,10601
neuronx_distributed/lightning/precision_plugin.py,sha256=mLgzUp5QCGCSBzf7IyejbIVo8Ke74PHmCuApkiEyMfg,905
neuronx_distributed/lightning/progress_bar.py,sha256=UBcIqrardU_y5rq8QWUnJ9wDzwZIkbAEywdQytjO4hs,924
neuronx_distributed/lightning/strategy.py,sha256=y5mQrWzq4Z88_fgHQb5LhrZZNm_-ktiZD0O3fNU1NoU,9435
neuronx_distributed/modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
neuronx_distributed/modules/qkv_linear.py,sha256=COkLaqbqOqm2R3dHOzPYdQyfh49FOsff59gWzm8V1ko,28857
neuronx_distributed/modules/qkv_linear_utils.py,sha256=NcceMBq031xrJ4Q33Dm9GZcYk84Lm2QKLd0N-bO7zo8,7559
neuronx_distributed/modules/rms_norm.py,sha256=-ilnbYPRvyYvrGAg32M2iyjWhBbiQ7wSmoeJnqU--e8,2449
neuronx_distributed/modules/attention/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
neuronx_distributed/modules/attention/utils.py,sha256=OhvsA7V_QH9CIUyPIehxADJEiFTgeyv2GPS_bCpsAtM,2930
neuronx_distributed/modules/lora/__init__.py,sha256=mFIgM_3RGcgCQv4ThPBlTq4Ij4HO5fJxzHGLFcUxfbw,450
neuronx_distributed/modules/lora/config.py,sha256=TOKWYM55zQ1GNvaIvKjV38HA_JayGgUI8G1aPaabsUM,7045
neuronx_distributed/modules/lora/layer.py,sha256=Hv_bw0d_iasTcHmBWp11jJUjtslr3FSi26KZ9Cf-jsg,17194
neuronx_distributed/modules/lora/model.py,sha256=oSkqdYHbiGwX9KsHljJ76EQQAVWOerTQwBqOcqAEJls,30552
neuronx_distributed/modules/lora/tp_layer.py,sha256=d9-ST4uOmkBGD_szckfSGf0o7bO9en4Jjj6t0wimNRI,7506
neuronx_distributed/modules/moe/__init__.py,sha256=lOJ2_OSEUfNoI4XPB3z4jtEXc22DYQIGrLmAzTModME,191
neuronx_distributed/modules/moe/blockwise.py,sha256=1Ou7ZAN0rmfJj9SsBrc0-sjLBUfjnBFDKNLmFQugrPQ,57375
neuronx_distributed/modules/moe/expert_mlps.py,sha256=JL-bQZqkC6q7NvialYUPSZFJ3sISiKk5V5592p2k62k,9575
neuronx_distributed/modules/moe/expert_mlps_v2.py,sha256=yOOE2Dc7uJ6802yv3Iq2NJw4Gdn7woeHs8xSCdJPLWc,88027
neuronx_distributed/modules/moe/experts.py,sha256=CEYFRDUdS5zOQy5Mn3kHCOMEdP3ST2UqfgZIKFmUpQo,10749
neuronx_distributed/modules/moe/loss_function.py,sha256=cEz2rAZBi4zMgm6o33131wHN-eTkvItm97jhX4hyHGg,2216
neuronx_distributed/modules/moe/model.py,sha256=l1AYrAJsIsgPjZZ6C1PkNFT8kUQ8p5Qi2RWSv4K3GqY,17642
neuronx_distributed/modules/moe/model_utils.py,sha256=JujnLHXOzaSSvdqQI85yhhwFvchf2f1ZUA6TDfBh0bQ,3668
neuronx_distributed/modules/moe/moe_config_validator.py,sha256=QevDzDX_PTemNItvcbITqbysgrqkURgHOx1ZBzr0MOo,5006
neuronx_distributed/modules/moe/moe_configs.py,sha256=pReeO4ZHFGeVogHw1i5LWk51dLNcxpk2uOj1WzksEM8,16501
neuronx_distributed/modules/moe/moe_fused_tkg.py,sha256=8w9XrMdAngeAmsa-JqABcK-UUYRY7up1_7t-ha4t5UQ,21122
neuronx_distributed/modules/moe/moe_fused_tkg_mx.py,sha256=MfCV6FgojnwsGYL3Gkp-AisGYVYt0e6kMDOhNz6AUCY,17125
neuronx_distributed/modules/moe/moe_parallel_layers.py,sha256=8rVGXWk1phOLnJvusbP-RcAeq4IhgJIuTPVRucK1c-I,17971
neuronx_distributed/modules/moe/moe_process_group.py,sha256=o9NWtUsrsHtOUWMHWerH-abqEg_I9qOvc2Dkf5IHpj8,4024
neuronx_distributed/modules/moe/nki_import.py,sha256=az2c1snB0a2aYmU0JAj8GOmDe3P9pK4xZpDuPXJZv-Q,3399
neuronx_distributed/modules/moe/routing.py,sha256=p2_cS7zDEKWJqrPRszH9AP-EA4kKVnUlchQ1sujk8yo,19171
neuronx_distributed/modules/moe/shared_experts.py,sha256=achQUqYcQYjq_MX1rY_pPYVPYNbP6hiMhP8i3hJb7bY,12628
neuronx_distributed/modules/moe/token_shuffling.py,sha256=UTomfY9dhrCP7joCuNL55QFh4HjeXWzyjdegcqah-HI,4088
neuronx_distributed/operators/__init__.py,sha256=uNoBSF_kUQAIBG1tP6Feh0IJy8mqqGJuoPeG4KxtNSc,27
neuronx_distributed/operators/argmax.py,sha256=w0sdB17x_ZTAVIstl0U-M1UkD7Jz3cZB8RL_q4_ND10,5013
neuronx_distributed/operators/topk.py,sha256=Z6_W5Vi-14xGNqC0rI7c5CmV3_PRsxwXjs_hXMwW1Fg,6645
neuronx_distributed/optimizer/__init__.py,sha256=C4tDYUc6pKdatS6734ae9J5OLW1WT9deFax3Gw3dPro,84
neuronx_distributed/optimizer/convert_zero_checkpoints.py,sha256=gVboqa2LfkqWjD545ewdEbujUcuJjcQpi64qJ2ngyS4,8451
neuronx_distributed/optimizer/zero_dcp_utils.py,sha256=1Ru1TLucbj8vIBwpNygSNba29NoMgXxuAngSdiwJn-g,19246
neuronx_distributed/optimizer/zero_redundancy_optimizer.py,sha256=fTjWQPqkAL_P9cMKekkqEJfk_XlqUKWrPpZBUt40Dbs,15927
neuronx_distributed/overrides/transformer_overrides.py,sha256=7GpR-lfYAflclZLagOYl86-Bo2bI0U5Xhg6Grw4dPzU,1295
neuronx_distributed/parallel_layers/__init__.py,sha256=rAFiy6pN_0Eubhv2LdGA18JqcYxRy6lJLtAU7gB1qRo,1430
neuronx_distributed/parallel_layers/checkpointing.py,sha256=K2WZgSqDEOq5A4z3jsQHLQEHP1oA2e3dJUoZEsty-aM,10607
neuronx_distributed/parallel_layers/comm.py,sha256=fRnkPkclXOPpQvkci9PM8KjaGzDHOrBtqh2qT1N5lJU,7070
neuronx_distributed/parallel_layers/grads.py,sha256=Z8Xz3EWedXzUFXM1fJGHr_-WK8PBmMfJdHeH0LUovK0,15160
neuronx_distributed/parallel_layers/layer_norm.py,sha256=gj4WrPL0pmC2v8qLO9QWqiQOAalXGIi8FEIUqQePhv8,1439
neuronx_distributed/parallel_layers/layers.py,sha256=IcGh8AH5Q5ef2LWkd56k9VsRvQbQu9IBwzsZnunv-7U,65807
neuronx_distributed/parallel_layers/layers_utils.py,sha256=L_CYZ0_u0-Q4l8UW-DKllqmMUAZ8Sh1CnFyJGOSoPUQ,5370
neuronx_distributed/parallel_layers/loss_functions.py,sha256=-5_EB46hphnm7yqyZl1qibCJF3ALms5hJbFoMTJkBQc,9982
neuronx_distributed/parallel_layers/mappings.py,sha256=8O74dkgrzKNuzTa51zROIkX30B3df7VjIXWRwZrpYYc,25758
neuronx_distributed/parallel_layers/pad.py,sha256=aG935GAJsfFuI_f9O2JthBKF9LWxVk5z9paPM0AF3Xw,8608
neuronx_distributed/parallel_layers/parallel_state.py,sha256=HbLQQVGqt4ePDfYhRxyxTtRZwSJaZfljAaD9-y9p-ys,73294
neuronx_distributed/parallel_layers/random.py,sha256=b7UmDS_GJ8RnAXcLOtlOmCCiPKlUa4aaIOV9mSOd6NI,4586
neuronx_distributed/parallel_layers/utils.py,sha256=kNN-RhWoSPIKsvQfu_JHD70LUl9d0Qa1Og3m22pqOWY,12437
neuronx_distributed/pipeline/__init__.py,sha256=JP2cjBqri7pbbzvQ7pdBPWPILZSLp-Rt9VONtQxBdCA,30
neuronx_distributed/pipeline/comm.py,sha256=X2Z0Zbvj9Z0JXKdQatRGb4Yk4FGV03ypfIq5xkBHEcg,7843
neuronx_distributed/pipeline/manual_pipe_stage.py,sha256=O6VriMULmudEOntzpv-C8Ov0sYHsfrMfrvC5BJU_GBM,12048
neuronx_distributed/pipeline/model.py,sha256=8rUgvCNeJjQrpTa-d9FBCGsodXQLg3bz5B3u-fdWnGA,97171
neuronx_distributed/pipeline/partition.py,sha256=W8a78pCBmoiO5yx0sM5PLnH1gUYwRuEF7ZTV5kqVJsg,13717
neuronx_distributed/pipeline/scheduler.py,sha256=Y8yFnUVzi-SLeKjKTNTVHpVPMj3Yy4q2OgqQkYa9MPs,30371
neuronx_distributed/pipeline/timeline.py,sha256=rgw9z1rqEflR26oaHCraATsu-H8bilN7LAH6Oq1SpNA,727
neuronx_distributed/pipeline/trace.py,sha256=hk3t_SMu9DeKAmmwkNpvXcvV5JrV4gEaznx5_wfZikA,8848
neuronx_distributed/quantization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
neuronx_distributed/quantization/dequantize.py,sha256=-3Q9hHvK2Kh26rmn2hCCXYmRqX6Y9jxj5csyHzhCf_o,5522
neuronx_distributed/quantization/observer.py,sha256=r5e1FziAzixlabiaG1hdO_E46OoH3jBMcp55NmdouNc,6207
neuronx_distributed/quantization/quantization_config.py,sha256=N0vIYtbnImenGm4Qj2vJ7EhVWYXWqo1p45W6GrW8-Dw,9695
neuronx_distributed/quantization/quantization_layers.py,sha256=stRzzZCRBz0YAF1EgmzF9Vq8zR7uFFk4FtGPGBHJn4U,67616
neuronx_distributed/quantization/quantization_mappings.py,sha256=s41KjmLeBoo_RJxd5PY81s3xkndxf2QnoMVDbdOU9H0,961
neuronx_distributed/quantization/quantization_utils.py,sha256=o-r5nxEijePAwUtN6W-fsZto7Ky3zSAZVJjUoyh864M,11082
neuronx_distributed/quantization/quantize.py,sha256=hfQcS97_KDEI3iJa13EFIcPn1epAbJez1nk_b9X-w0U,7029
neuronx_distributed/quantization/microscaling/transform_weights.py,sha256=vb07kTLY2QV7LV5ofIv4axlW7YKIK9o_vLUIP2Q_cOE,14209
neuronx_distributed/scripts/checkpoint_converter.py,sha256=WHh3oY7Rkarb-sdVZ7dmQ2t8QYJ1zlg93pPq24VX044,51820
neuronx_distributed/scripts/yaml_converter.py,sha256=jBCLKa75A7ozZp7MUHQl4C6KPz0QnBkxDMN9ZiwTyh0,1558
neuronx_distributed/trace/__init__.py,sha256=61faUEd64rGNGpENw1Ew6Hkuu1gSqZ-MM1rqiBZ-Ptw,184
neuronx_distributed/trace/functions.py,sha256=IP0QbBLiNhmpzIaSjD-1411UcqHa0UzFaSltI4Md_Hc,35565
neuronx_distributed/trace/hlo_utils.py,sha256=SCrethocv68JXq6X6XO94EtarAJrrGnar-kC0dRbfXI,33716
neuronx_distributed/trace/mock_torchdist.py,sha256=zQsHtQw5DJ8msdDwmUAyM4bb6OTEOQNdzjAAbI7UfsU,2612
neuronx_distributed/trace/model_builder.py,sha256=sq0SiFcf5I-C0ATYbY1zb3-tXI8HARdbW1orIeM41pI,63001
neuronx_distributed/trace/model_builder_utils.py,sha256=42lIbcR21nR9VRAXgaYjNCJlEH0Jezniw85mMsSS4Qs,5374
neuronx_distributed/trace/model_builder_v2.py,sha256=62LvoxcaOqlZwIAamh-cAUGpvVBwXu-VxNVljfX8wgM,12223
neuronx_distributed/trace/parallel_context.py,sha256=SNcGnjOinSJKr2mNjgzkwXYzt7TuZp5MpDOP3ksHAio,3848
neuronx_distributed/trace/spmd.py,sha256=xVhVF6wJnjo06GNyg7vbkXBUjPI2v2yTBFGM6JukHlw,11870
neuronx_distributed/trace/trace.py,sha256=xa_3_HAKiVBbdpgxTn-drzUnBXrFCQtQobNUGwdL7y4,34514
neuronx_distributed/trace/nxd_model/__init__.py,sha256=E159tJYjtq4r9VgqJGL8WTBId7MF7D6xRzrj1bzDa3c,134
neuronx_distributed/trace/nxd_model/base_nxd_model.py,sha256=puKOjp0_hnLQT0LPkCAg5RXcrZE_Kk2-25I9VyCmbxE,7342
neuronx_distributed/trace/nxd_model/nxd_model.py,sha256=m47EjN-LFgPZfNdYXT--wG1d5YaccQ6jbyyCGb54JxA,44197
neuronx_distributed/trace/nxd_model/utils.py,sha256=Dj0THjlU9jJiMGS9LzUM-QnHzyYiWxiYlr8u0WOhsZY,6712
neuronx_distributed/trainer/__init__.py,sha256=FlOHfSPZZoUdW6N4aXbfU9lAcdLvYLpn1NVaWrdfVS8,353
neuronx_distributed/trainer/checkpoint.py,sha256=YziTalsnpsnPPbMbvADYqbcaeb_SbFTCGNmL94N3P6g,40161
neuronx_distributed/trainer/checkpoint_storage.py,sha256=IEdvJkZZYgC9TvGHYDEW3_YI3bmuxUhNjIpSgC0ii30,24059
neuronx_distributed/trainer/model.py,sha256=2imJay-CAM4FcKShQt_HrCwR1D-gCEVzzJw1399UhKE,3584
neuronx_distributed/trainer/optimizer.py,sha256=4a7zWtcBC0omAZgCwOuz5wn6XTwdQma91ALafuWNBX4,6235
neuronx_distributed/trainer/post_partition_hooks.py,sha256=4KEyD6PDtUcwq-8FS80O8UC2i8aet6z8kcXHzpEKtM4,1152
neuronx_distributed/trainer/trainer.py,sha256=jD5OpNRzImJh6E2_K7B2wifNaN8IbT5QLUKVi9opX8s,16983
neuronx_distributed/utils/__init__.py,sha256=g6GG4xo0f4YB8noe_fhZUqbRWMbZXQsK1L6Hn7AGTIM,522
neuronx_distributed/utils/activation_checkpoint.py,sha256=bFPmz7wiMaE0NLkxRJtfEAle2xieTYz4SwRgjLX-qDs,2904
neuronx_distributed/utils/adamw_fp32_optim_params.py,sha256=cLPR9zFErJ0Fs5_ZaygXRpIEP0x1H8zh6EWMSaVEgl4,7033
neuronx_distributed/utils/batch_utils.py,sha256=Jwxkune4Hl93bCXo6mvfCXhyZ4ihBzS_9nJtjD4ax8k,1729
neuronx_distributed/utils/logger.py,sha256=h8zSJoh7wU_eyeAi7lKaL05JwDUGkpJ2JFGtCPQOOJs,4128
neuronx_distributed/utils/medusa_utils.py,sha256=oM_LU61qixQAdKrFZlK6Il1S69fSvthHmFfd-0inyZo,8713
neuronx_distributed/utils/model_utils.py,sha256=ef6BeJZJy-iZs7wm_uKIbY1QujXDjycSftS78m06Y44,14365
neuronx_distributed/utils/random.py,sha256=ndUDhgALUqb_SA79h3OW0vmdUZ8-pjkLPwtR28Dgk0M,379
neuronx_distributed/utils/safetensors_utils.py,sha256=JwJaE23qenwUOTO1HDeQnnfKFxi6mbB8bsCr0Kme8os,6459
neuronx_distributed/utils/sampling.py,sha256=L3UloLwCzfS5tifCfzl5dthhoP6DufrXI0xcactYkzA,3375
neuronx_distributed/utils/serialization.py,sha256=H1I-NPb7bJHbcAbXkcCJQ4xwNhERRbRcFNrXshTlTlI,9768
neuronx_distributed/utils/tensor_utils.py,sha256=nzHscJVmWjEfAmqHGhlbEELDpdVqX3kcPbxvY6mYPbE,2520
neuronx_distributed/utils/timeline.py,sha256=T8GwdC2TLFa2qSMmXDoBwqo4HHatqN3W-jNLwNz0m2Y,4072
neuronx_distributed/utils/utils.py,sha256=r3nnB_JhbdeRGK_ggQ0OaYkPxFBKR-eO5Am8YX_OQmM,1976
neuronx_distributed/utils/tensor_capture/__init__.py,sha256=rgDAV2sPZlqk1l2XhTFnjUwlnZ2eErI40kJF7Uzzk20,576
neuronx_distributed/utils/tensor_capture/api.py,sha256=jhUZVARROCB8VxghN6X-V_RGvxoljuO49GDJ4KtLqhM,2840
neuronx_distributed/utils/tensor_capture/model_modification.py,sha256=H9O6xL3SOfAg3MS0fPKXqSDrPthb9T-_LCOeZKRz9iw,10635
neuronx_distributed/utils/tensor_capture/registry.py,sha256=U25DVIwhJbYppPDNWDv1_6bZbx_jy2imFZF1KccCfhA,10687
neuronx_distributed/utils/tensor_replacement/__init__.py,sha256=tbs4x7nx45qZ57LqniE6o7Po8Tv-eiQD2YEgaCQE468,518
neuronx_distributed/utils/tensor_replacement/model_modification.py,sha256=jxYpEJrZGidXWT3U2YjydMSyOfT0QFazONFkpq2ROSo,2896
neuronx_distributed/utils/tensor_replacement/registry.py,sha256=p196AKwinOBig2skhUtWI1J_66Zgzxx0aGcEtpVZ7tk,1336
neuronx_distributed-0.18.27753+1cafd54f.dist-info/METADATA,sha256=hrCmj3Rr5bgNPk88q-yc0dtmKsHjgn3m5R2xza4ZTns,1201
neuronx_distributed-0.18.27753+1cafd54f.dist-info/WHEEL,sha256=mCaSw5Ewi7J7lmDh2K5epnoqNFHn22lOMDyuOGA4Eu0,102
neuronx_distributed-0.18.27753+1cafd54f.dist-info/entry_points.txt,sha256=IR9om-dLyKz6drq5cQW8iIFimWlcNz3s4VU2JisaD5U,109
neuronx_distributed-0.18.27753+1cafd54f.dist-info/top_level.txt,sha256=tnoINeJuj78M85dE1w49rqGVHN6BL5nItxiktsrTUIM,20
neuronx_distributed-0.18.27753+1cafd54f.dist-info/RECORD,,
