apex/__init__.py,sha256=Xi7XDImSgeaVXzfl3-GfYdJdDZaprNSmcGaT65oU_Qs,2563
apex/_autocast_utils.py,sha256=qaxNutZvdijaLApAHkTypuK5cl43Kq0bgzPLxYFZ3Z8,690
apex/RNN/RNNBackend.py,sha256=Tp-absO-WvxM8kozgu9axhU-5G0M2MHQrXOC8TSGT-g,11938
apex/RNN/__init__.py,sha256=o39PFXmEkp3IVDbH1EO4aWc5djd4wDehaC6kHv2_INo,74
apex/RNN/cells.py,sha256=aqyo9kJOd7_4Kz6IREgYYsSYK58SU0l4A2Xj1IAVxKA,2632
apex/RNN/models.py,sha256=pVEcU5z_T2Kp8YPNKKG4H6BYKOaQUmz0Ft4LnqlN-DY,2321
apex/amp/__init__.py,sha256=Cl1gPfdCkKSasbtUk7puxabZJaen3ig1qWSzPcybidQ,315
apex/amp/__version__.py,sha256=LuRuM64HoWn8HuFAFj-XVvEsnSr6RL-Aa24B-gATpI0,64
apex/amp/_amp_state.py,sha256=kA8aiJKMG29n66YfMqPQEb3kP0pBLQSlJrqoJmvS--s,1818
apex/amp/_initialize.py,sha256=e8P1pvSlBH622mn4gTovmsbfAEFTdXWJFli6y6gTPKE,11849
apex/amp/_process_optimizer.py,sha256=CZdTCHKb8tzJg5NdFgH0VTTfAfIzU0vHeEKcBN8gWj8,21236
apex/amp/amp.py,sha256=N6sOHyXaeOc6VSOvKRFqrfthbcpMiyB0_-krI1kKDRk,8043
apex/amp/compat.py,sha256=-BIK5XlI2g36rQ2RZtTlNDoh4pzi5056cHUl9tVoZTg,1439
apex/amp/frontend.py,sha256=EXNHCFvQH96Z7pTZsjXLuh39XXMz8YGbvRCFdtPfj1M,21909
apex/amp/handle.py,sha256=K1wNDqUNBgiSTdF-iCQWhwUc1YhEOnXsPFQb-E-QVu8,12347
apex/amp/opt.py,sha256=QaQgsDMMkTIQxuKKP2WLumEm9KO2ALAVbj-XttF-Cz4,3549
apex/amp/rnn_compat.py,sha256=SVO7B6KoQaeawnm0eLuSBUz99gMWya4hVhPHsYXwWN8,2048
apex/amp/scaler.py,sha256=kJBW41ako_77dyMfJhMKSPWwnYnqyOrJpI23gJ37yfc,10711
apex/amp/utils.py,sha256=hEiaR3DAhxG5wu6uzVTLdWyHroH07tUtlmqd6T2ezqg,7432
apex/amp/wrap.py,sha256=S0B1ASt0mjBVX3JwLJS2zMRkw8IFm4jsU1qsXwW0ag0,11518
apex/amp/lists/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/amp/lists/functional_overrides.py,sha256=r7PTAz-xsKpaYo0LKJE5LSStbdq73n1wb0CizFcIuhU,2328
apex/amp/lists/tensor_overrides.py,sha256=YVlCnE-_urSDzuvShNp1YFZBy6Q8xO8sLHHS_wtrSPE,1465
apex/amp/lists/torch_overrides.py,sha256=XkCXOVu35UyiMgOBQCOLwisCNe-6OdakL4ClYneKOyo,2197
apex/contrib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/bottleneck/__init__.py,sha256=aWJbKmmReqiElXgQyUzvmKbTevnmMCJRTmExvckDYxY,171
apex/contrib/bottleneck/bottleneck.py,sha256=Zwcw85H-ALFsmqCdyUOPpTRTFgZf32JFGhOy69Y7qUE,37307
apex/contrib/bottleneck/halo_exchangers.py,sha256=zMyiRyiFF0aXDtUoSKVvahszFF3womvwzvnhS_dzwdg,9933
apex/contrib/bottleneck/test.py,sha256=M_L_wHdvgPHIjYIqxQqESD4vqo0lsL9dJZ5fbUqONL0,3141
apex/contrib/clip_grad/__init__.py,sha256=NQpBVc_ScDqx0IFHJZ7HAVVkNrQe8j8pG-5W-dObjdY,40
apex/contrib/clip_grad/clip_grad.py,sha256=a3cx7x5ovyzt-Ub4b-nN1Mu-ukJWHkvZLuiD5WJgJuU,4488
apex/contrib/conv_bias_relu/__init__.py,sha256=85Bgg5ZvHNiNr2cLyqGMJjKzZe7YGG2nqAyQSOwZVA4,97
apex/contrib/conv_bias_relu/conv_bias_relu.py,sha256=Nt_yyUvHrGTfn1-3x-SWITkNX7LsjoirDtbs3uZPH9w,3560
apex/contrib/cudnn_gbn/__init__.py,sha256=pRdT9QyPVeAZdpMPWTh-FnMR9AIrSLJDZmvF0n8955s,40
apex/contrib/cudnn_gbn/batch_norm.py,sha256=6qkinlwFkliU0K073cfQpvzwxhCQ3pREyMcCIsf3GnU,6869
apex/contrib/fmha/__init__.py,sha256=3Va89pVbG75a3GK4IEl0Ag1ITTogmCdxhAqDB7LfXN0,27
apex/contrib/fmha/fmha.py,sha256=3ZTWYV3nG1AdAHdamVwN-OKCbwfc01d2Mw9GBgd8bwc,3653
apex/contrib/focal_loss/__init__.py,sha256=0IsCA-EresskkPD33VJGEHd2uZXTnH9U12d-Zo2bIfg,281
apex/contrib/focal_loss/focal_loss.py,sha256=4DGNnLyLD7nsA4eXcXCz2nvTQxHLTil3K5FgzRRrzf8,1559
apex/contrib/gpu_direct_storage/__init__.py,sha256=Oei4jW4fDMy4lxaO495USr4Pff3WLtRVhMCyNvowdw0,352
apex/contrib/group_norm/__init__.py,sha256=-jjoH5qbjDqUyFgiaMAN0cIXauvz6aEu6YGzeE3VPyI,27
apex/contrib/group_norm/group_norm.py,sha256=lwhH-eGFC8_hf5r4HeGJ8lc7Zg8TaN2eJRrVUySzJc4,15408
apex/contrib/groupbn/__init__.py,sha256=kF0iUaVe7jngoW8_cvdioX8L4k2OiZcNgT51_aaaLAw,248
apex/contrib/groupbn/batch_norm.py,sha256=LuXwwsli_a8D6rShBzwLf_3w3sWXVvc1ODy8clp3wsg,11433
apex/contrib/index_mul_2d/__init__.py,sha256=JrCenG360WLAo0eczcquOj6NlkDM9fWu8L75RN9Ijvc,40
apex/contrib/index_mul_2d/index_mul_2d.py,sha256=17wSkQCFP0lgv3XSNyZKoZqvch1CthDCZUlnzW0aHLs,4738
apex/contrib/layer_norm/__init__.py,sha256=nj14NoO_7WZvqBapmpMCT9GA5xl0murYvyZswgbdT10,39
apex/contrib/layer_norm/layer_norm.py,sha256=QQN_hoZeaVMiLvBiBw5f52neYHz91f5KADoFq1xQxBI,2144
apex/contrib/multihead_attn/__init__.py,sha256=U6c9HR4Fp9bDmBi9SoPgOIY0tQk0arnW4nwFFI_0edg,179
apex/contrib/multihead_attn/encdec_multihead_attn.py,sha256=TkzdvOOp_FLQWchxokMIPYADyzJB0Ui7iBfz6k1akMM,7726
apex/contrib/multihead_attn/encdec_multihead_attn_func.py,sha256=jRQR--b3QoDwrqxTb9rnCQXYIGrolgsixJdPy5kIfU0,17200
apex/contrib/multihead_attn/fast_encdec_multihead_attn_func.py,sha256=5Z84u2ZH7bWCWQa-IbyyZd5xyBTV4PBCObl3LqJ9r6s,3095
apex/contrib/multihead_attn/fast_encdec_multihead_attn_norm_add_func.py,sha256=qnICTMhcT9LvO-USm_oPXuUNJptg3gwA2F6_orMkauY,4417
apex/contrib/multihead_attn/fast_self_multihead_attn_func.py,sha256=lZ3wbJWVtOWWTJnvM19oj-RBWovzunr_Nv30AFdflR8,7922
apex/contrib/multihead_attn/fast_self_multihead_attn_norm_add_func.py,sha256=YETI6NSAXcwVdalgpS76QQ8yT28gEDkIUsbeFilx7gI,3627
apex/contrib/multihead_attn/mask_softmax_dropout_func.py,sha256=dAA_EANKzQykUVreoiF8tx63crw6bRye7e2jZ9Wv_qA,2520
apex/contrib/multihead_attn/self_multihead_attn.py,sha256=ZvXMs1uxsGjV5ggQlC063Om_gy2XqDA2LZfXS6qs_H0,10339
apex/contrib/multihead_attn/self_multihead_attn_func.py,sha256=rnwO95cezaImAN_WEhzuwroyL9UoZbn7WXuhwfo7vlk,14452
apex/contrib/nccl_allocator/__init__.py,sha256=-Y7e2fxvHqhSk3CB4XiQJHEuzW9Ru6J_OH6kYFKoB1E,31
apex/contrib/nccl_allocator/nccl_allocator.py,sha256=07nAWG4RcXl3nvBXPPGKzGHyElSCjEz-KoqSkC5fBAE,1962
apex/contrib/openfold_triton/__init__.py,sha256=jUhwiD5IJSFLmYyEN4f8rKS1FjFDbqlCeljRsvjb5u4,3899
apex/contrib/openfold_triton/_layer_norm_backward_kernels.py,sha256=oeFkwqpjO5b3bOorQZIiS_2UJHVPHxJETTQfnJYxREs,10889
apex/contrib/openfold_triton/_layer_norm_config_ampere.py,sha256=UtWImgbpgCrs9g_LhFJtmyPllHrlAnsIdw5yGpQymeU,6127
apex/contrib/openfold_triton/_layer_norm_config_hopper.py,sha256=FZBaGykuROYlhmF2T1ZCrWFH5XvEbjqsm4_iY9pztkY,6130
apex/contrib/openfold_triton/_layer_norm_forward_kernels.py,sha256=Xn8q1PtVRtqpcP7f-y29vtSkzNxtsSQNjMXbAl0_-tY,4054
apex/contrib/openfold_triton/_mha_kernel.py,sha256=_bAGpKU4FaJYGvpnMZ8vL5YCf_mwPbRgMHFWSlD_ZpE,22205
apex/contrib/openfold_triton/fused_adam_swa.py,sha256=Le94J2h3n5LIzphnPFHYJC2VwcQmu-N2cfcQNpXTqw4,20196
apex/contrib/openfold_triton/layer_norm.py,sha256=Zz8UkjboJkEBvcRSzLPtqpRy0b0NHY9Wc6bbueIwX5U,7162
apex/contrib/openfold_triton/mha.py,sha256=ALbLQpXa78bV-VdC_3iLBDgv1V1pOrP1bbGU0hgXxb4,14717
apex/contrib/optimizers/__init__.py,sha256=CHJRqCVm7rRcPsH9MDjIpmLIsEMmQ0AOzm9AEhOYCmg,114
apex/contrib/optimizers/distributed_fused_adam.py,sha256=FJBl7_GGS8jAlGoeu-GbXdZc-saVrT_TS3gWbJ_IFbg,150103
apex/contrib/optimizers/distributed_fused_lamb.py,sha256=XvH5TCfCJS1JiFG9EC33Vk7i3YYUEdR8zVPrEoXJJYc,58041
apex/contrib/optimizers/fp16_optimizer.py,sha256=tP7heYJQPKy51MzQOA1vbBWsh75ktPsRoeFUAf3q7aQ,10691
apex/contrib/optimizers/fused_adam.py,sha256=UgKuHoleN28tFJI45vZhhkTY9U5DhqewhxwTJ6wxAsU,9490
apex/contrib/optimizers/fused_lamb.py,sha256=4tnFWRcS64d9Ty5kUkb8FaB4bbpk_L6lqTRMa2UbRjI,9616
apex/contrib/optimizers/fused_sgd.py,sha256=BhdhalI8zwUJfncmn2fFki1DUSJ7nX2qEz-SYQA0-SE,9679
apex/contrib/peer_memory/__init__.py,sha256=Lu3BAbbOO_k6NuMD8u5xKdSo22b1Yfy-E-ueE-YJF_g,100
apex/contrib/peer_memory/peer_halo_exchanger_1d.py,sha256=KzlVVKThWn2psD2IcrUleLj84TDzEpHqPJVVIElqgnw,3923
apex/contrib/peer_memory/peer_memory.py,sha256=lXdBSyNR_dh8yTu8hZ6dYviXlKvk3-NKhjclcknltD4,4835
apex/contrib/sparsity/__init__.py,sha256=ePHM_YERaWO_fdPMfwMxUeOfTPP_P6vHIlSpLN57Ztg,63
apex/contrib/sparsity/asp.py,sha256=wPXmuEEc8LdUpXDivnbA89FzLNE02P1vRoRzVTubkKw,18767
apex/contrib/sparsity/permutation_lib.py,sha256=IWORyQnxX7clSiwMpsBGpr3_ToHqjRIxn3bfj83nSEs,90872
apex/contrib/sparsity/sparse_masklib.py,sha256=Wm2XRhTAW6zzteojmNGsJWaI8syEBf0O5OO3o6MRzd4,7609
apex/contrib/sparsity/permutation_search_kernels/__init__.py,sha256=SYwxI54vVca3NPoYI7dsEYZH94sB0BKu2nXk33ej5co,137
apex/contrib/sparsity/permutation_search_kernels/call_permutation_search_kernels.py,sha256=d_iEwqn809EIKAbakvB6ZWaNYJCzoUSm8xdsqAC-MGo,3985
apex/contrib/sparsity/permutation_search_kernels/channel_swap.py,sha256=8DNasjHwh-vdtj2K8_4GyxNB3n5TgRF-EkRailQ6Zac,8729
apex/contrib/sparsity/permutation_search_kernels/exhaustive_search.py,sha256=qRoV-6FZKHu4x6EoFK4XjJimtj0Vn-NNMoruNfbaHno,17954
apex/contrib/sparsity/permutation_search_kernels/permutation_utilities.py,sha256=oqsYg1hm0OrhBd-W57OqmEu0nIFS_5ybsQz_pMycqy8,21997
apex/contrib/test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/bottleneck/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/bottleneck/test_bottleneck_module.py,sha256=yFwsSQ88je96hEjERS2JJ6dymM_rW2-XVCAEo2C2RX0,12030
apex/contrib/test/clip_grad/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/clip_grad/test_clip_grad.py,sha256=V6IRL2NLs0S0OQvPrAi1f5LIumYsSjk5S2PJ_FAgqt0,5155
apex/contrib/test/conv_bias_relu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/conv_bias_relu/test_conv_bias_relu.py,sha256=AmIVXifdcqXDoJZn-oanQ093g1ETf-RxGJwQA_nXjg0,6725
apex/contrib/test/cudnn_gbn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/cudnn_gbn/test_cudnn_gbn_with_two_gpus.py,sha256=b7X9U1C6U4ogRw107HinHVny8DL_app5t5ElgvEhbhA,5050
apex/contrib/test/fmha/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/fmha/test_fmha.py,sha256=K6kda7ieyP-mvYvd4WjQFiJii364WmuPDHgHHbd64oY,5573
apex/contrib/test/focal_loss/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/focal_loss/test_focal_loss.py,sha256=gWbfmx6o4ydhE_n7xMwmr5s6MuiIq4DS5t0yGdugE6s,2327
apex/contrib/test/group_norm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/group_norm/test_group_norm.py,sha256=fN9HeuD_Y5L9ZtQV6YTaYRx69FAR5XOJ2xelGm1ikAM,14622
apex/contrib/test/index_mul_2d/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/index_mul_2d/test_index_mul_2d.py,sha256=r_ya54gIC8Z9y6VpStvO_H7zHqWQQTVTiPyJlHMM18o,4441
apex/contrib/test/layer_norm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/layer_norm/test_fast_layer_norm.py,sha256=Ef2xBjrWNMvpwLJ_3ZrDQluw_iXuWocBju8gIGqdL6A,8750
apex/contrib/test/multihead_attn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/multihead_attn/test_encdec_multihead_attn.py,sha256=ei_H9FAluaWaEx04NRBuCrKVs9TskFESafO00oNYQEE,6051
apex/contrib/test/multihead_attn/test_encdec_multihead_attn_norm_add.py,sha256=VrivCBiXgz37AP4_lYV5pUfV36tivmLJkHAE782BNlk,3241
apex/contrib/test/multihead_attn/test_fast_self_multihead_attn_bias.py,sha256=QiaAY-ucjwIQF3BE9pVD3rirKstGivTPy0uBX2VqYA4,2991
apex/contrib/test/multihead_attn/test_mha_fused_softmax.py,sha256=AgK0hhSEdmRHJTeldHOelLMdV5XKOcmiSoGZIjU5H8g,1954
apex/contrib/test/multihead_attn/test_self_multihead_attn.py,sha256=fUHPLoaRMwSrv1CPen0k_mwpTT38-bEcVMfb88b6_Ug,5343
apex/contrib/test/multihead_attn/test_self_multihead_attn_norm_add.py,sha256=yRs_cVbEJmYzCRftdYnR986kTwPPO50mC3ztadIlX8o,2694
apex/contrib/test/optimizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/optimizers/test_dist_adam.py,sha256=Qsm5c6-sJEdnDBQhHhjYity4Vfk1vSPR-o_KWm26etw,33556
apex/contrib/test/optimizers/test_distributed_fused_lamb.py,sha256=-mnBqH394EnlhT5VbzRpa3xmIXKgU83qsh3g8NzeURo,5138
apex/contrib/test/peer_memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/peer_memory/test_peer_halo_exchange_module.py,sha256=w1hOk32xGmNhUZBCnCSN7ENhbq5dL_X8E4IHG3pbRAE,10459
apex/contrib/test/transducer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/transducer/test_transducer_joint.py,sha256=IAfUMTGxPC4I8HZZqRze2D98d01XAso_OKoI7j18vFI,7384
apex/contrib/test/transducer/test_transducer_loss.py,sha256=JthjTvXmgmXaFuViwPYiAMdKO4YJmLM_RpxLLpCPS8g,7061
apex/contrib/test/xentropy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/contrib/test/xentropy/test_label_smoothing.py,sha256=NZRlkeNMnq9Faf9DXdWPrvjq0ZD_G74bTsfHSOh56NE,5179
apex/contrib/torchsched/__init__.py,sha256=hDtvxKNAKFc3grXkco4cxMxi8ys_IlM1XyPP1E_jwF4,2386
apex/contrib/torchsched/backend.py,sha256=viewzDdqiHkigig9CTfMc72Ua-yLQwH5w139__qjtNQ,11507
apex/contrib/torchsched/config.py,sha256=NLMuthQVGTuQBITDjYcv-T7qX42hP2KQLd97s9iHT5A,744
apex/contrib/torchsched/inductor/__init__.py,sha256=DzE-DtfgHhnGQnZhLcTWwSxSO6AiMKWTIOUy6gmzUCk,171
apex/contrib/torchsched/inductor/_utils.py,sha256=S-QzNHcPYhcJrygVppg2uPN2w7zSJCcOGhOo6eHeYgA,4222
apex/contrib/torchsched/inductor/event.py,sha256=m9HQ1mnb2sTm3AJvq7MpxsY2MVIDz74lJ1ggbNNdUkI,7587
apex/contrib/torchsched/inductor/graph.py,sha256=Ji2MTvs_bWT1dVRxLcUkEUatfw6FlMlugxnJGATno5U,2158
apex/contrib/torchsched/inductor/scheduler.py,sha256=BCyNY3HcUHc5Go2bmTFJ6frPvi7k426BPh-VyAb1nQ4,26873
apex/contrib/torchsched/inductor/wrapper.py,sha256=wLtXen1K2jD23OA6wyQ0a5trU5isJQhQf37RluORmhA,13272
apex/contrib/torchsched/ops/__init__.py,sha256=KdfDBYpduvfgbbqSMqbtdjDpfSfqeyPia0QY2WNLeq8,174
apex/contrib/torchsched/ops/layer_norm.py,sha256=Wdx-BxW1i-DzVv4JYyvdNkV-59HGxLNMEIEMfY2he4M,14645
apex/contrib/torchsched/passes/__init__.py,sha256=VEex32QwFa75bTY0v4bovAgpJ4w_wWTv_yl3Ger90wU,194
apex/contrib/torchsched/passes/pre_grad_passes.py,sha256=lZ5fO2UgWXwknh3rA1bN_tA0Si7NtVQHrdxwGuikvKA,3453
apex/contrib/transducer/__init__.py,sha256=5pySN_45v6NtEvPabUpl5CDG_NFR_4ajyXSotIEfHaU,112
apex/contrib/transducer/_transducer_ref.py,sha256=X3nJUg-ZTNk8wPCjnKhJ-Uc4V2bwKrwy28T702id9Q0,4730
apex/contrib/transducer/transducer.py,sha256=dp_u3fTTLdn8iIvHOMGwac3U222tL_T4GJFsXADaIvo,10129
apex/contrib/xentropy/__init__.py,sha256=T4DbNCFQv83lMCpPWNJfE4CpPtW878DzjyRBoUU-tw0,107
apex/contrib/xentropy/softmax_xentropy.py,sha256=_Cc010ExtoHNZS0THWwD0xbAFF0GHcgKHmtKXfXSzDE,1055
apex/fp16_utils/__init__.py,sha256=V7sTnhkGKlOQvU-QKspze1ONeOAb4nWJIIhhRteEiV0,383
apex/fp16_utils/fp16_optimizer.py,sha256=EmG_SQ7uMMEJb2S9P8wZwWwfQ-ovogHkHxxOYDsCUt8,28438
apex/fp16_utils/fp16util.py,sha256=UzMc43pWk0IR7Ks3PRhc_LodMnjiUj1qztK71Q96zaQ,7540
apex/fp16_utils/loss_scaler.py,sha256=4Zi5gBqn7lR93LUWsQs89u_i1xp-MprVif-qngIwuCU,7928
apex/fused_dense/__init__.py,sha256=Im3jQnoD4UIg5A3uq6ytKHuArN8AxdTRIG-z7HXVy6g,28
apex/fused_dense/fused_dense.py,sha256=XMKNyVzIjYOt910sHtmwt6tsmtbwGhJFc-nX8QT3qmk,4176
apex/mlp/__init__.py,sha256=Dh_s9kGbVMnIH0kZYcA5jWCccIsFeBuXkargjLP0eaA,20
apex/mlp/mlp.py,sha256=TyJaCUyManFvb1C3bfwrBxw3T7YlP7aF_0FV--hKouU,2843
apex/multi_tensor_apply/__init__.py,sha256=mYJnPSollKiayipRRXfTjmDkrI0yWHnjJAgfNsDDP5g,104
apex/multi_tensor_apply/multi_tensor_apply.py,sha256=lKxTS9hTjU4o8Lj_MzD4jMNN1FEKlNwhdU2fwrWAN50,1021
apex/normalization/__init__.py,sha256=cQE8bp3iHDtylbP0SEY1Jl8dZNR2YbHksZX9kgS_7MI,100
apex/normalization/fused_layer_norm.py,sha256=tCWsHPfXddX41bb1bciNG_XHCJrIIuoD_XuaM-Q6hwE,37624
apex/optimizers/__init__.py,sha256=9V1deJjlKvrxqVm3LUJg-eluSrScQphnvwPaiOS3AUc,252
apex/optimizers/fused_adagrad.py,sha256=nAAVT38sEZ0kEl56_BYwrBwI1R3UncsKuhmjQlWwiOs,5352
apex/optimizers/fused_adam.py,sha256=ZRQXxn8WobwUouHSkBQ2SKNUOrNVzy3pXAHMljscwk0,14270
apex/optimizers/fused_lamb.py,sha256=oQN8KsVosvEFrOu73kp035LyOWkuNoJqOIhhBMH0HnA,10125
apex/optimizers/fused_mixed_precision_lamb.py,sha256=8xhNfO38D1jgw_bpoEKbDHg_L8bMmKaneDn8xUcPiEM,11534
apex/optimizers/fused_novograd.py,sha256=s9Hy01KPNRTG9rO7X2RLeKAsQU8G07tL0k2-_1yWkck,10866
apex/optimizers/fused_sgd.py,sha256=BhxaXY81kOKuyZmrezhDduhNCeY7XA3R4I2Ft1c8q8g,10268
apex/parallel/LARC.py,sha256=ImjroH_FefzlpKb_eYUGUyfye3lG9gFN2q6TSvi5Evc,4125
apex/parallel/__init__.py,sha256=ra1frXjEGhEC9Y6hCeSxJKw4lUm3l3BtMnjzhPp76to,3981
apex/parallel/distributed.py,sha256=_HECscF_eQb9zwNVaBEb6jfBjpeCwttCiHM54w7dxYQ,31471
apex/parallel/multiproc.py,sha256=KUTEkLcqx_MhV5dCbeIa39jL1Lxz570CCr8E_tWCVuM,919
apex/parallel/optimized_sync_batchnorm.py,sha256=Oyb1SEGDl64tzcEn2LCWwV8zrPQlpySE3sUvG4ZLGtA,4449
apex/parallel/optimized_sync_batchnorm_kernel.py,sha256=yaeOtX2TIJsG9NZ8ZtTQm9u-X2-JChJawXOljyANgNc,5586
apex/parallel/sync_batchnorm.py,sha256=NCrJKgUOoAksA1fIM6wSBo6wdVCGkc0dOdwcqhVavsI,6863
apex/parallel/sync_batchnorm_kernel.py,sha256=gAArXBc4_mCwJUgIIpalsDanmavjqSDsYFhWAy3ywmo,3848
apex/transformer/__init__.py,sha256=Ut2hGKvrXfnkS7Mcjl1hv1xz3XNcdaAoVHiaqAbsOeY,600
apex/transformer/_ucc_util.py,sha256=Ilxj2on5EOYB13p5A1iBWHY80afRraY9EVqgQLhDX0Y,241
apex/transformer/enums.py,sha256=AZ-_dSWVB8YIdrWEEyplzrN-HNxDRB7lKSKxUGEaNeU,949
apex/transformer/log_util.py,sha256=CoPpM79OWBDFqeV-tsr8nfFaLrtvxCPMHzYZDsQU5MQ,389
apex/transformer/microbatches.py,sha256=WMDOCo1YCIXgwy7yacpw-8aQ2ahFnOcAP3Wn3jfzpKQ,7368
apex/transformer/parallel_state.py,sha256=elY_AXkhNfCOhkFRCL2_Kyoz797DYfFinHqJxUUi5X0,33866
apex/transformer/utils.py,sha256=0xHgm71iRyCOnp-OCKJIAPEkjoRAGWOZhl4LM5Wmu2U,1977
apex/transformer/_data/__init__.py,sha256=uLJiU4QVPieiIVXKo0kbaP9Qnm75vX7uTrznZY0rny4,256
apex/transformer/_data/_batchsampler.py,sha256=M5z2197qQd6G0HimjqqyjcXDwd5JHPeZTZDo-SS6--I,7383
apex/transformer/amp/__init__.py,sha256=CtNcdMYKu39ZrYk6JTOp9zRZv-zT-fKhGJO5gQuUGJ4,96
apex/transformer/amp/grad_scaler.py,sha256=Rlx2RSbMPi51nt_T8uRxTZvcsiq1v7jO7F5XgPMvWs4,5702
apex/transformer/functional/__init__.py,sha256=6HQooTfhC-5MBPwuFYCnQlltxQS-IIhTXq4qTeot8v8,482
apex/transformer/functional/fused_rope.py,sha256=1nN5fR5PX4CH8g5bd-TE_QWVxOq8OaBVv4UrmaSi_00,10075
apex/transformer/functional/fused_softmax.py,sha256=55egGTCL03bJwkektSjrCph6zh5roIevvyIxFtL64zY,11227
apex/transformer/layers/__init__.py,sha256=6iiJpRUBwILg6LLHi2spoTw1mdxdNRGWnDnovRH0NmM,350
apex/transformer/layers/layer_norm.py,sha256=G7_0fvHzlehQ1THHAM_XMtTWG6w5RELAwFii0t9qrzg,3555
apex/transformer/pipeline_parallel/__init__.py,sha256=1cD2BN-3zC0HGPX-z_i-LSJ3OdlfWmg4jnhnn1kIAq0,235
apex/transformer/pipeline_parallel/_timers.py,sha256=OzRYENkUn6p7Xz00YakHKnoKtxvuvpE8rVza0WElhVU,2621
apex/transformer/pipeline_parallel/p2p_communication.py,sha256=icIQNqmVtzr19IgpwrpSmfdosMgRyZYJmWtNADJRNeo,29024
apex/transformer/pipeline_parallel/utils.py,sha256=wQYEkKM_9vkOQb7kXgUqorogH54TEy2_1P9SP6xR7hY,12967
apex/transformer/pipeline_parallel/schedules/__init__.py,sha256=PBwzSmyryz6CJ67r8JUHcf8yaXOnWKMOQUKLYBhh2nY,1434
apex/transformer/pipeline_parallel/schedules/common.py,sha256=5PjxNonWrCl_rcex__POG1k2xHfEGMneO-zioNaR0w4,16235
apex/transformer/pipeline_parallel/schedules/fwd_bwd_no_pipelining.py,sha256=0VYHBDreqqIs6u2PmF1Vyaadmcq62T4pzEXrzvP8L0k,4787
apex/transformer/pipeline_parallel/schedules/fwd_bwd_pipelining_with_interleaving.py,sha256=ow1gvGX-sInT6yTN7GoW4xPAifAXaRR865KsWsoKCUA,36026
apex/transformer/pipeline_parallel/schedules/fwd_bwd_pipelining_without_interleaving.py,sha256=LvOzdobgfw3dY8N5UgWWgDvTLZK00jKyYvXBHvh78MU,24972
apex/transformer/tensor_parallel/__init__.py,sha256=SUpg44CCqJdA5XEtxUiYncLEqSutHjYbocRBMTDCXwE,2589
apex/transformer/tensor_parallel/cross_entropy.py,sha256=3J8Ie6H6kkzUkRg8aibnYZ6mIrKLg5Ck5DLlt5zL4Do,6580
apex/transformer/tensor_parallel/data.py,sha256=Pbu5t7VcCxTYs8gB-GNUrutSaBsnKWSA8_PpMmN5UvI,4169
apex/transformer/tensor_parallel/layers.py,sha256=GPZTy3trF7ijf7AFqjSWLnvB3HhwWiycO_jcao4Jtm4,31837
apex/transformer/tensor_parallel/mappings.py,sha256=RvLdj_aR-q32wRBdFPcPvRh6kmmZ0_DRiDPYAU8XhTA,11230
apex/transformer/tensor_parallel/memory.py,sha256=0drPj3DDbs0dNLigyysEeLATJMLi5eHaSzzOf64IcDo,5354
apex/transformer/tensor_parallel/random.py,sha256=Sfhu-Auw4op4RrtaOS-kDO-uiVF7X6gM8JQU1G9o9rU,12163
apex/transformer/tensor_parallel/utils.py,sha256=Ccm4ouHHf6TgPcB_s6J4-FsVeC1bRUViWtmaJdvMkOY,2460
apex/transformer/testing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
apex/transformer/testing/arguments.py,sha256=L5vuBBAgOPLksJ88Gzx-VSKJYG0sx-mRroLrZMl0xmc,51626
apex/transformer/testing/commons.py,sha256=zAnJqBI4cRi5vL0qW-BOhHBaa1n3yMzlTPJQ6kcr2OI,9978
apex/transformer/testing/distributed_test_base.py,sha256=gW9wc56Zq6q9sA9c8_nN8GfvbvdHQM4xLL68WH7-XFo,4301
apex/transformer/testing/global_vars.py,sha256=nIOpFKqP_Nw5Gx9Gaq5UPnXcL2uDaEM-GylgT9F59DI,9281
apex/transformer/testing/standalone_bert.py,sha256=s0tUtNsRxScN5R5ROP7b9I1Gk9k63pQ4kPRW0w5gDxk,10200
apex/transformer/testing/standalone_gpt.py,sha256=aMa6vyrTXEyvIOw0T9cFZEOZQdHI8CCDp6qOcq_QiD8,4046
apex/transformer/testing/standalone_transformer_lm.py,sha256=XV_bvF0onZd3kxOp7HOMCzXR-n09Lv8mFtBts2ba-CI,62389
apex-0.1.dist-info/licenses/LICENSE,sha256=zk0H362MvTFBWIB7k8OZG6FlxTOh3aBvNgrLuAsZEL4,1459
apex-0.1.dist-info/METADATA,sha256=5GuF6-iu4ozia0v27t_odG4K8cfpavoDwp-LOECqH7M,215
apex-0.1.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
apex-0.1.dist-info/top_level.txt,sha256=hXyJoceHr37HA55Q9iQDNNvPW59kaSlaJlIRGvBuvuY,5
apex-0.1.dist-info/RECORD,,
