torchtitan_npu/__init__.py,sha256=QeUSlnOESe15iLJ91xpqHpeg4IytKpH6wTJyXLMMf88,2487
torchtitan_npu/entry.py,sha256=0hodsaypdR_9zshkeU1UnXsz7J8eEzlCnvgeZFqKi40,4554
torchtitan_npu/train.py,sha256=90uTDP9kk3kfj_q_Ib8IKD-RqveKvS9y_UBX68R1q7M,8745
torchtitan_npu/config/custom_config.py,sha256=6j_dbvqI6gkg6RBu-5pRJCDXkKTRcXXEOYcg8kn_LlI,7313
torchtitan_npu/converters/__init__.py,sha256=lTv3ciJGB-29zAHl2AGCVU_gwi0lPLvvEMGUYqcLGF4,918
torchtitan_npu/converters/base_converter.py,sha256=4kyE511sl8iyGCEn-0c1ddhBdOPN1wsCASuyr-M4ld4,1112
torchtitan_npu/converters/convert_utils.py,sha256=oamHSNv-wmQnAdoWJZwjjJLvNb6H7UX5beeSABZEG0g,4702
torchtitan_npu/converters/npu_converter.py,sha256=SDgsyg32AWn42wcis5aUi5Q7m32F35moSMELiT8DY_8,2183
torchtitan_npu/converters/quant_converter.py,sha256=yZ_9a07H4vC3_ELgMfPA1Kim0RxCDqItqNrP2tPX-oQ,4874
torchtitan_npu/converters/registry.py,sha256=3dXzYIHpM9OarrhkOPd4DZGYbr8nzIMWib76athhrfc,2172
torchtitan_npu/converters/features/bypass_triton_codegen.py,sha256=Mho7lxYiL5ThcJQAIJNZ5NO9GRzKrlLql3XkzVtBCdA,2094
torchtitan_npu/converters/kernels/deepseek_v4_sfa.py,sha256=2kFd4sBH7zVKaxlyBsDvYnAecEq83ofgan9t0k2bI2c,14320
torchtitan_npu/converters/kernels/dsa.py,sha256=DNTLswmGVOO7rPA6TDCoJz2vo0BwfTQcerfJ0Nl3j-0,10426
torchtitan_npu/converters/kernels/fusion_attention.py,sha256=aiAL0dgx021ykJ_Zwdy1jjgZGYynOrQzKM5xDg-F_34,2275
torchtitan_npu/converters/kernels/gmm.py,sha256=jQDU7cF3U_B9Vi2y2Nksp5vXuoBMulC31r-YKOJGD2I,8029
torchtitan_npu/converters/kernels/mhc_prepost.py,sha256=_BDQXVCrV0YKmeAz8bGtyRWvLOKxyoXeNntN7YtuwTU,6138
torchtitan_npu/converters/kernels/permutation.py,sha256=H7HOVga8BnXQ-6liaGr2uo_P3_mpFO_YE9D3YnIkFYw,1826
torchtitan_npu/converters/kernels/permute.py,sha256=UBNbWPK4-32dCcM_gSou0t_qi5OrSRrTOhtQ6rptUuw,9576
torchtitan_npu/converters/kernels/quant_gmm.py,sha256=nqitrm1pyfltLrjt4zpZfySyv3Lw5MjBdiPB7S7-72o,6637
torchtitan_npu/converters/kernels/quant_linear.py,sha256=aATFAqn0A_aDfcspAinI7UNs67axTXieNbIfypINjC8,6426
torchtitan_npu/converters/kernels/rms_norm.py,sha256=2YtbBAVz_rPVZeoYuFV2XTwnmYnFIfpxfwQ3l58PRm8,1998
torchtitan_npu/converters/kernels/rope.py,sha256=seic8zpZMeW13dFimLzpadzMfgG4kXuNtVXpmqCCGes,5917
torchtitan_npu/distributed/context_parallel/__init__.py,sha256=xSkKj07N7mHg5q2Ov4NmU2RWLUxZ1ar0T6bGhRxlqag,495
torchtitan_npu/distributed/context_parallel/custom_context_parallel.py,sha256=bm1UoFWSOxaTcECJSErLCWdn_gegCkHz1-NdkzXFqvU,2213
torchtitan_npu/distributed/context_parallel/deepseek_v4_cp.py,sha256=MDI9nZRCRMiqyOvpAcKpFnD5qG71TKCUtArDrioW_5k,8441
torchtitan_npu/distributed/context_parallel/dsa_cp.py,sha256=YYIoSIFuc7hi4p3eTaQAPsUzybjUUbRikG0DR7t65wY,9500
torchtitan_npu/distributed/context_parallel/ulysses_cp.py,sha256=10lSzN8SI9lP9hsHyJR9qnOzNmdEgsH86hRufUjWwrg,3927
torchtitan_npu/models/common/__init__.py,sha256=CJhnavsIaD9kIuPi4UHi2IBhmFfb3GxbY1bZaoqiUss,204
torchtitan_npu/models/common/dsa_indexer_loss.py,sha256=_SGQDkma1lFjUC692BzIDU3huTeYI5Ix_aac8xxSfLU,4919
torchtitan_npu/models/deepseek_v3/__init__.py,sha256=sIN2KrksUhSHoh_MyCQnHGGQII4NiaC5RyJ8JYNwFfA,2981
torchtitan_npu/models/deepseek_v3/infra/parallelize.py,sha256=ASvzYd-i_F_2b4X3JPZuqHp0IqQ210sNM94Ih_h-YbE,2091
torchtitan_npu/models/deepseek_v3/model/args.py,sha256=ZXvH_uRxAvho_01VZyd7h5yOrJQTnzdDDfKfkKUeVF4,673
torchtitan_npu/models/deepseek_v3/model/state_dict_adapter.py,sha256=P_hVPdRR7QNWnQR_xtg65vwrVoLcywUzmQBh1eKKsPo,1810
torchtitan_npu/models/deepseek_v32/__init__.py,sha256=x9n4uQm956dH5P5VsyvbTITK7PWDMgDds5VCq9HIbJA,4861
torchtitan_npu/models/deepseek_v32/infra/parallelize.py,sha256=UcU527ADNPM2nqdrS9jS4K0Jtge-sswb68v32Q6Cu38,32390
torchtitan_npu/models/deepseek_v32/model/args.py,sha256=2whNSNM4M2KsYFyH-a493cSsX9fqIl2wTDL6g-8kZBI,1185
torchtitan_npu/models/deepseek_v32/model/model.py,sha256=IMB_FqaQsNd7NZtASXF6syc_3Wt8ZQWkbSMM16iyQX0,28144
torchtitan_npu/models/deepseek_v32/model/state_dict_adapter.py,sha256=QpNr7M0ysm2IhTq8Fbx0I77zTKnm8MA7WWZmXnZuJk8,6418
torchtitan_npu/models/deepseek_v4/__init__.py,sha256=y85B5C0TStN2BGlYEtmtS3nfLFurFnEmP5m0yc0k5iQ,7179
torchtitan_npu/models/deepseek_v4/text_datasets.py,sha256=JA_gcQr4C1eElljypcDTGImURLOfrASjqnFXEk-62GI,11635
torchtitan_npu/models/deepseek_v4/infra/parallelize.py,sha256=SPsYEGlMGcZIMZk9z2-P5VX4c_kDfbpWaOEImRBLgJQ,41075
torchtitan_npu/models/deepseek_v4/model/args.py,sha256=swVg1rlLrGqE5iL19O_SFL72o4Q4I5BjPadIjcLG7gQ,3796
torchtitan_npu/models/deepseek_v4/model/model.py,sha256=lXozU-iTnqAj-J_dlUa_UliDMAxgAJTBI1s8rhvq8-E,47798
torchtitan_npu/models/deepseek_v4/model/moe.py,sha256=mMQELReNgXg65wIv7V0FWnmS_xSXYopRMZXqtGF3gus,10803
torchtitan_npu/models/deepseek_v4/model/state_dict_adapter.py,sha256=oLjhEmGZepSHgF0Cle22NNEAPPzrtY6GcGHqwjqrGck,17726
torchtitan_npu/models/llama4/model/state_dict_adapter.py,sha256=-kKQNuxp7Sd7ZleWW1qPDPgneR1RWf5ZAf2D_OHdeq4,2737
torchtitan_npu/ops/aclnn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
torchtitan_npu/ops/aclnn/builder.py,sha256=mPjhz3aBFXtRHVkgXbSlYWs98vj_W02KQd7hOhzW15k,1538
torchtitan_npu/ops/triton/__init__.py,sha256=a867UMv8bXlfyKAZXJdZZOIfv_TrzA7fnu6Wuc_pjQ4,135
torchtitan_npu/ops/triton/add.py,sha256=HASZeKYnhMMp5NBeERd7RbmKK6wsIbLLKME28UVCF2s,2716
torchtitan_npu/ops/triton/mhc_triton.py,sha256=vt9qIVTdYZICChLMzpF0hEfvC4PdqkB8lW64o3g6MNE,9389
torchtitan_npu/ops/triton/post_bmm1.py,sha256=JIHAEvUN38vhktOZ7_l2RkSY0DRz0JqGNz0AQCc4rJ0,9182
torchtitan_npu/ops/triton/post_bmm2.py,sha256=TT3ZuueJsZ6jCAehD-3le87GHECjRxFf7dnN7iJRf7c,14373
torchtitan_npu/ops/triton/pre_bmm.py,sha256=XPCWxgJ4unSVqXVeNCDf96FMxNj9sNuVVYJaJiE59p0,8753
torchtitan_npu/ops/triton/prepost_sinkhorn.py,sha256=_e1pwrjLmw-5qMaPG3N4BkLKuxbohcBUg5boc3F918o,31179
torchtitan_npu/patches/distributed/cp_input_sharding.py,sha256=gglaKzjDdIbiSFUkfDtl2pBqejoIdl15BW7bCe6S4Z0,1871
torchtitan_npu/patches/distributed/custom_context_parallel.py,sha256=3Gop7Jo84HtwH9NNqh9QU0-Yqqpx8iOCafEnDykQRjo,5177
torchtitan_npu/patches/distributed/mtp_context_parallel.py,sha256=36zrtAeot0DPred9aqupEcDbAn_roYpw9TUG41lSzjY,3796
torchtitan_npu/patches/distributed/utils.py,sha256=A7pRddhvSWanLAFPIFzuB5BEwG4NcK4yU2XDfNPg4co,3152
torchtitan_npu/patches/optimizer/muon_optimizer.py,sha256=fyRNS6ZJU_cxAAOoQSO6PPoDIrslb97-cIQihuYBvIc,64865
torchtitan_npu/patches/optimizer/swap_muon_optimizer.py,sha256=qEGq_gp3p6BvS79dnx36J7iXE3bZfvNpo9Gv3xt_Ea4,20841
torchtitan_npu/patches/optimizer/swap_optimizer.py,sha256=2ITLiBrqTttvPDsaXAiqtR5ixoDBkT3D7VtK409c-tA,27429
torchtitan_npu/patches/optimizer/virtual_allocator.py,sha256=vsRH2oeFAjBE3iHt_F2lu07B_z06RkERiNNC8WBgjvE,4731
torchtitan_npu/patches/optimizer/virtual_optimizer.py,sha256=0JrF0N4Olt6GFnMD4ML1e2Kzp3CsYKjLFpKIuS3t7Pg,15470
torchtitan_npu/patches/quantization/quant_config.py,sha256=QppJ4pyE61TlAoLe1Gw5ffA_XlAcGNIEPJOKxrJVorg,3177
torchtitan_npu/patches/quantization/quantize.py,sha256=T3xCV52McYM3QpNYGbZxrtjNIeFF68DzU7z0wIZgguU,4167
torchtitan_npu/patches/tools/metrics.py,sha256=vdJt9bnPf2VTZUMisFXYuDbpSBeFD2y78SdVfZiMY_o,4945
torchtitan_npu/patches/torch/checkpoint.py,sha256=jYaVRxe9_mmtERXosK6kSVwNZDb-qSoLMPOo7xU7mog,7374
torchtitan_npu/patches/torch/clip_grad.py,sha256=sGm64Q4ZAW2gFfzH6IxzSwpCRchZzfMQPZ5C8g1nG10,6164
torchtitan_npu/patches/torch/micro_pipeline_tp.py,sha256=FbskwEImOoPD5-mO1YpLe6F0uMA8mIBQU7ZvuAc8h9Q,4847
torchtitan_npu/patches/torch/pipelining.py,sha256=r-Iwq7HoYk7bTML9CTazux7O-j3C_BaaGyV1sfoYqCM,4164
torchtitan_npu/patches/torch/_inductor/graph.py,sha256=BoJM3BVueQ6XkcBZrBpOd9n3AmfYl-A7IduaVHGxMzY,5009
torchtitan_npu/patches/torch/testing/_internal/distributed/fake_pg.py,sha256=QM1AO1o3j9nB2rDKmLE0wBU1AcJfyqawTHwWgxU6mmc,1356
torchtitan_npu/patches/torch_npu/_meta_registrations.py,sha256=gMIEeELtC_vbqWi-sEAIlI4i8vTyB5LtL46_REVdGuw,1760
torchtitan_npu/patches/torch_npu/custom_shardings.py,sha256=sy5lB7rrQS-0qx-dIqxs4NjanhIP5Lf7ekKzgUOm_zU,11209
torchtitan_npu/patches/torch_npu/_inductor/lowering.py,sha256=i7NM9Y0HPdKmM1-KciLXHfIg4JaA2PWEaQjhYTEUaWA,833
torchtitan_npu/patches/torchtitan/hf_datasets.py,sha256=lZfw3i_hTcRlQAYm5XihWMyRNKSO6fJDaE-OQ7FkkYw,3676
torchtitan_npu/patches/torchtitan/loss.py,sha256=W9P2B0vSIfKe50ju03ZkQ_XN3isEHTLd5UmIi_IFDx8,2198
torchtitan_npu/patches/torchtitan/lr_scheduler.py,sha256=gJ_ZYv_l4F1szNQd2o1_Uei6iOtHHS9szJV1o-gF8zo,2850
torchtitan_npu/tools/checkpoint_patch.py,sha256=UaReR2zHKd8Ik7mz-9IYdyDn9fO7grOmC072Q5D3FWc,12332
torchtitan_npu/tools/flight_recorder.py,sha256=3B6roz8lMIiffLugwdhZmvMEd9jt2JHQLnLdFP35M-c,1771
torchtitan_npu/tools/profiling.py,sha256=pDDXawt6dRAHF-60xKKyx1XTJD1RsORXEg8qsKYUv2Q,5080
torchtitan_npu/tools/weight_utils.py,sha256=Pbh_wb2LGHtNMpA3Yc8sxQPFnmXQ43GiDRuxat6JeUU,6100
torchtitan_npu-0.2.2.post2.dist-info/licenses/LICENSE,sha256=CPSdYUZ5gXsJwinamcxLU8IW5eJ7YUY-NOLCiXOPDO4,1565
torchtitan_npu-0.2.2.post2.dist-info/licenses/NOTICE,sha256=N8tCrvtMazslSpwjy62HONlVT7209Xioh6eBieNqcjU,5328
torchtitan_npu-0.2.2.post2.dist-info/METADATA,sha256=jrOS0jWwR_ErCRqu8GOQrHBnfIXOekrnj6Ga1JFNLks,14330
torchtitan_npu-0.2.2.post2.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
torchtitan_npu-0.2.2.post2.dist-info/top_level.txt,sha256=uzmTrOJOVs-NME65MDrDyrdX5EimdVxcVV5XdBnR-c0,15
torchtitan_npu-0.2.2.post2.dist-info/RECORD,,
