transformer_engine/__init__.py,sha256=JwceFnnzkN87YNJJ-H8Z6KUIzpYq8NvpNKsQQ8-eqdw,466
transformer_engine/common/__init__.py,sha256=V4Aapkvg-7TQihpTAy-7luVaEdDu_lvnXXYbsmWb6uA,4509
transformer_engine/common/utils.py,sha256=jwsON-BLagT8lPr966PwmapbsUyDMy6BmyFpK0z6mlM,1646
transformer_engine/common/recipe/__init__.py,sha256=1a2CPp-iyuGSr3-srxpLmSl1kIjSj62UJecdoICg2As,17593
transformer_engine/debug/__init__.py,sha256=7Cz0u61rSC58Hp2LVUD0pVoPwaNzAo80J2gN36e-DNQ,309
transformer_engine/debug/pytorch/__init__.py,sha256=reUoLQOU3obVMuEs9PACIOu9ql9pPST1QCvM_C-DzSQ,122
transformer_engine/debug/pytorch/debug_quantization.py,sha256=DIKqZkQPGE9salIE1-Zn7G_CHfOlIg55GnEgqwubFX8,21478
transformer_engine/debug/pytorch/debug_state.py,sha256=nmj96nJaq-5wMYEO1Ch5bXeC4n_UGXRofKDff2mDg5A,2128
transformer_engine/debug/pytorch/utils.py,sha256=JusIjqtetNrQMQn_fem5pVkhYw4yOelG5dfOBamL_5I,343
transformer_engine/jax/__init__.py,sha256=z2VXHdY2pRLWoZcogR2J4uDxllKAW5-VHCGrX9WzOgI,4122
transformer_engine/jax/activation.py,sha256=rlv66ENkpJg4NyrWUwwFYpj6mM9E-tey6Y9n5oNd9lk,2846
transformer_engine/jax/attention.py,sha256=OTNm1oiTNeLwhdptQ6byRBzsQ1U1PrWKihPte9RjHmM,43908
transformer_engine/jax/dense.py,sha256=B0yM_uor1bmnZo8rOhFSLXKPRvidPM7bU6O3vThMrno,10907
transformer_engine/jax/layernorm.py,sha256=0kyNSbjUfVzgX_ljf74Cffmn-5QJxONNejmnPp9AHWU,4454
transformer_engine/jax/layernorm_dense.py,sha256=YHgPr9SCo-mJa4SnwU7CSQA8VesvOVx3HYHSL9fGtUw,9272
transformer_engine/jax/layernorm_mlp.py,sha256=LTgkV6_nVD7l2Zwo5MLt7IBTnbtt24fet1swUBPyDfc,15958
transformer_engine/jax/setup.py,sha256=ZzBmqCrwKQ-7ycWYtpi_kN1OK5Z-cqUfBELPWNWKkT0,3462
transformer_engine/jax/sharding.py,sha256=pRs8awRjWySQDJWSLM0IICjK9GKiPketSCoip1exMFc,10514
transformer_engine/jax/softmax.py,sha256=IujIGDvRzVJQ9bm3lH9PqwuM77aOljcfkNtZ7Wdf3xk,2051
transformer_engine/jax/cpp_extensions/__init__.py,sha256=mSg0rGposfADwGsl3a92DJcWdmN2DZ-CceVeSE7vHLQ,315
transformer_engine/jax/cpp_extensions/activation.py,sha256=SBoO0amchNKISoRGqmIwmG_DSmZ-6FoYsNo-lBRF8M0,39870
transformer_engine/jax/cpp_extensions/attention.py,sha256=zmqMSewPf4_Ll_yiaiNbl27TwOHGpjgZlZ1wwS1VFtU,102997
transformer_engine/jax/cpp_extensions/base.py,sha256=e3kst886eXQwXPjKz1_IaaVgi6nWjEMd3YXIXok-J3M,4116
transformer_engine/jax/cpp_extensions/gemm.py,sha256=29QqQgnFhNFGtcS94WfIMOUGunNfSqmxmJ2EpVB_jEc,18536
transformer_engine/jax/cpp_extensions/misc.py,sha256=xfA_f3Hdm8_rNdWplanl7lGIADxeWtP311FE-S0uc2A,9037
transformer_engine/jax/cpp_extensions/normalization.py,sha256=PE7Odgradr7efQ3_rDV3xuvWcfVN8wAM9OnR_BH2-CA,45594
transformer_engine/jax/cpp_extensions/quantization.py,sha256=5ctNU_h-tztLTXFDvSu_iFrniUU8zEmWX0cAa81fmL4,22531
transformer_engine/jax/cpp_extensions/softmax.py,sha256=d_gIiw9nmqh7ZbZFB_XZcBC7Iq0_aqq_IPS6ny3cx6s,30420
transformer_engine/jax/flax/__init__.py,sha256=wohmcM50EsahqVjO2IowXtGhO7IhGOVUS9HmJvoJz7g,737
transformer_engine/jax/flax/module.py,sha256=zN3qigmWGFh8A1_PfooZhahxwbmuDNjmpixXSLXJtaU,52006
transformer_engine/jax/flax/transformer.py,sha256=EQsH9ggKt7oAmS-jwOHfgsbAFetZaYniS1a5ClN12Ks,92884
transformer_engine/jax/quantize/__init__.py,sha256=j-g3rSZF211AhI5T26jnaAKEIGbt0791amtU3DR5T2o,567
transformer_engine/jax/quantize/dequantizer.py,sha256=bQr2tm_Dqb7ywfPv5n_m5JdA2IEsrTqS5L5TSrgyBcs,3879
transformer_engine/jax/quantize/helper.py,sha256=XFpaD3eYuxTC3HDfNWuO7QPK8-6IIcoTmJBlrX6Wyn0,15441
transformer_engine/jax/quantize/metadata.py,sha256=mUcgaxEldPjekCuEEJr1Qjn1du-DudwoHoosQvZJxu0,1038
transformer_engine/jax/quantize/quantizer.py,sha256=Mi5W_KDGZt8j9p8Gg5QpomyXWdQ3gyH0YmcpkDI54II,22656
transformer_engine/jax/quantize/scaling_modes.py,sha256=ZFv1FoTHeBls17wcDzvVp1J5ysyZiRsZ7ymMuVHAqwM,15637
transformer_engine/jax/quantize/tensor.py,sha256=2gqTaVbz6cnX6AOEToHYaYJMkdISHs8BW3hwyEIMlHA,15610
transformer_engine/pytorch/__init__.py,sha256=Dkc3RRvzNvsMRW0oIO2pIID6vVWiHNdGr8ZG14hVpWM,4598
transformer_engine/pytorch/constants.py,sha256=ZtT-1WhUWmNBLxD2P2rtZBUAE7E-MdaVY_uZFXgGPtc,2091
transformer_engine/pytorch/cpu_offload.py,sha256=6Qjs1DPmRtyBKdbOjmPHuiSMUEBlbBmnXqqL-8D6cUQ,25261
transformer_engine/pytorch/cross_entropy.py,sha256=DC2FBVxWKJiJ0-rTzUSSldO5P4ASPvi7pGLn2OOCycw,2711
transformer_engine/pytorch/distributed.py,sha256=-fa4GwAVP57Oub12ywo3hmN8OKi0fQbzF0Zw_E7Lj0s,57998
transformer_engine/pytorch/float8_tensor.py,sha256=KdHMbjsSpPWkB1Cu4d0Z01QhATMijK03e_XCl1p88fo,232
transformer_engine/pytorch/fp8.py,sha256=hQiC33lkXkjgP8tmA9-XLLaftyH5rdl6b6pWEyPeWpE,39605
transformer_engine/pytorch/graph.py,sha256=q3gfJvGa7JJypufyMlVVZKBC1-MCIyEGYtXQcYFQBrY,33869
transformer_engine/pytorch/jit.py,sha256=O2BsnHal8WfpS2OYIjYpFLhG7D5Gd0VXApVOCeQP3ZE,9273
transformer_engine/pytorch/numerics_debug.py,sha256=dayXGDf0_qDtPBhRQQxMIdsSFzdkRmaeUF0PVNVmGaQ,908
transformer_engine/pytorch/permutation.py,sha256=2Kf1KtIi5u7SaEGKYg0WSy4b_uigzRSi0gohEBVO_Ao,28664
transformer_engine/pytorch/setup.py,sha256=qdLFWpteS0iGgNWzi8DAJjFJinZBWI6LMlyXzP9rgLs,2014
transformer_engine/pytorch/transformer.py,sha256=nQkmY28GOS0__prM2ph3MSqZyDEzi7aGf5ABFnZOfI4,40029
transformer_engine/pytorch/utils.py,sha256=j8872y5TEqCHJKlieU2GB-BjPmAyPXhbbdAIUqXfGhY,20587
transformer_engine/pytorch/attention/__init__.py,sha256=FumPWLRiUpKuMebYNdd7NCXg6tudT4tBHARDJJKFZKI,472
transformer_engine/pytorch/attention/inference.py,sha256=F1fQKystl0LEQSgzeH1WSGnJML5uGseCb44qUlnuOrI,28772
transformer_engine/pytorch/attention/multi_head_attention.py,sha256=0QSOck5lRVJDPsJEOQXomarv8sy3BohA0ZBX-q4RwNI,40340
transformer_engine/pytorch/attention/rope.py,sha256=-WrDRQ4Zo8JIunTp7iJ1LuNwTXtAvmexKIQjC1Ni2gE,14406
transformer_engine/pytorch/attention/dot_product_attention/__init__.py,sha256=A1MgBAGLR2GybJIGOIxw9OZjs34Bbq_4sb7boxTOiWo,307
transformer_engine/pytorch/attention/dot_product_attention/backends.py,sha256=LlMjwnlNUxrQHlQsfSTrxacznM6dvDx6jFx21QJiMVc,70499
transformer_engine/pytorch/attention/dot_product_attention/context_parallel.py,sha256=sUHslsQwkD5qAshj9gcGHdiGvE_IiU3Z9UcPx72opOA,162957
transformer_engine/pytorch/attention/dot_product_attention/dot_product_attention.py,sha256=_o7BimtHkxw0Y_BVO8eDguCeEsjRwoIG_1XE6lBH-P0,60643
transformer_engine/pytorch/attention/dot_product_attention/softmax.py,sha256=m-xaEG89MZKmejE2c4KVFiXO4NnrAhCOJ5-MetyHyQw,10244
transformer_engine/pytorch/attention/dot_product_attention/utils.py,sha256=w2K4pc1eX72rNBNgg4sINaeKALFhMuwxYA4NQtmXnWY,77922
transformer_engine/pytorch/cpp_extensions/__init__.py,sha256=qZ1qHrZN_Du1jhkefQx1PtQFRAGsvPrpOR-b5uUhoIE,251
transformer_engine/pytorch/cpp_extensions/fused_attn.py,sha256=lU6a-N-NCMP1v6y0uvPhUwL_vH0kW4xSbg8c-1Xszj8,20079
transformer_engine/pytorch/cpp_extensions/gemm.py,sha256=vntKlIe4GjYZlLNyAPuSZmsAoJZbtBA6tG3bxzQv7mY,5829
transformer_engine/pytorch/module/__init__.py,sha256=JHc9_LIv9Uoe4N-vyiGxS5GspAu9WWl4fNWyVQP8mAE,506
transformer_engine/pytorch/module/_common.py,sha256=vfAFrTFWJhL-Iib5_ekdeAs9eTucdFf7nlCVvfmj8Kw,9816
transformer_engine/pytorch/module/base.py,sha256=dQz9WceWBmazRb8nZdvXf7E0lGE-e72QYvG-L2PZdfw,52315
transformer_engine/pytorch/module/fp8_padding.py,sha256=rEPmiUGrb5K_Pe2HqOsV-uRgoQ0NlQMcoShpw9DVSVk,3805
transformer_engine/pytorch/module/fp8_unpadding.py,sha256=kVsNzsoXRYHNRkhShAn2XFoXC199PTAFQIU-XCEay7w,3775
transformer_engine/pytorch/module/grouped_linear.py,sha256=lJsHeOchsInr48kaRnEJg_m4a4wZHQHHO45FFknw4K0,34101
transformer_engine/pytorch/module/layernorm.py,sha256=E11uH1YpKsR78YneQ5LtWVbHD95E0Ps7S4KKvqQ8lT4,6542
transformer_engine/pytorch/module/layernorm_linear.py,sha256=2WLmvPT5G4tRiwV_UY9izJfFTTh82QqhWJsp6nb4Xxw,68376
transformer_engine/pytorch/module/layernorm_mlp.py,sha256=yuXlCxaX1N7Jxp7ndDu2_QTW2oUpn1qIVJVYkh08YOo,85111
transformer_engine/pytorch/module/linear.py,sha256=YD9eiRTJ8YqSHyG5TUy0WyYe1Cljg9UDhq-LJSbQQBQ,60518
transformer_engine/pytorch/module/rmsnorm.py,sha256=G1eVMC0VTxKGaeWbPjrRHO2gQIhI3hY1jXAWf-tLhQ4,6528
transformer_engine/pytorch/ops/__init__.py,sha256=3TCOnORonjWt7x1eW7pg2S2ZRoYC4Q_DGPWkArRCpko,453
transformer_engine/pytorch/ops/_common.py,sha256=1zTbwUauZKS5rkuARLP_qWGsNAqgMblzrpM0zfTtKWc,4217
transformer_engine/pytorch/ops/fuser.py,sha256=JWX-ZRRRfqtURn2h3bAJ__y3Zh__DwwSna3PVmavgWQ,15225
transformer_engine/pytorch/ops/linear.py,sha256=EElBJ20M9ncn5LB5vjOXzWDQm6aJKPDyvEAncUkJUiY,5882
transformer_engine/pytorch/ops/op.py,sha256=ps3U2PfJf4Vv7sGLLiPOZfBeY43Aus4l3CdVQtyjHbg,25362
transformer_engine/pytorch/ops/sequential.py,sha256=Jv_hYM9b6qPAZhbmSzw7Wg7zWGv1mosFLHh_kXvPDco,6718
transformer_engine/pytorch/ops/basic/__init__.py,sha256=ixT7L5NlU4Xda3-7ZdhS-FIu6wAkC5Y3Xx_UCa5agmQ,655
transformer_engine/pytorch/ops/basic/activation.py,sha256=-znUxVPHPjYRLS0yDQ-v3K81h7E7MMyB8RLNfAFptK8,9316
transformer_engine/pytorch/ops/basic/add_in_place.py,sha256=d89w6z8uAtNT-C9hub1aR3e6IQakKfkpQ69r69sQpZo,2605
transformer_engine/pytorch/ops/basic/all_gather.py,sha256=tWjo8JKmR8mXdcfL3NXcUYhNhbjPaZOq_BFT1GtJgoY,2583
transformer_engine/pytorch/ops/basic/all_reduce.py,sha256=ni2IFYuc8cNEqk58S7a_F-VykWGUk0BvhbcJaELvZkc,1847
transformer_engine/pytorch/ops/basic/basic_linear.py,sha256=q5_52VQu4EccHpBbIREc-yoX9SsFLaN1GNG8qBKq254,37407
transformer_engine/pytorch/ops/basic/bias.py,sha256=5Vc2iSlq7Q0EFvK-AWyG2yLYnF4ubC407EXuYgr9ixA,4247
transformer_engine/pytorch/ops/basic/identity.py,sha256=mN5wMchakX5ooGRO1Wo8RtjWzAA-nzsAiLebctkEo4E,811
transformer_engine/pytorch/ops/basic/layer_norm.py,sha256=w36MZhIAhfuigfHKzqq8wkF-vYXw6WHsD-mzeI7Wa4M,9630
transformer_engine/pytorch/ops/basic/make_extra_output.py,sha256=qh-1CP9sroJMHmdGzN4afEDWlDFBSdnQTbv4xo3KKh0,2727
transformer_engine/pytorch/ops/basic/quantize.py,sha256=Mj1GJvxKPZjXf_xGGpOVQqI7D5Hl5k9b32dtvf75u38,2275
transformer_engine/pytorch/ops/basic/reduce_scatter.py,sha256=UsaBzWli8pnLpRGEKzYFX_WK1IVqmE4ioJWpwm-1HgA,2546
transformer_engine/pytorch/ops/basic/reshape.py,sha256=qGf3UL4IbczF7f-dkXDmcl3IDDsbf2SJt-ca-Z5S-00,1251
transformer_engine/pytorch/ops/basic/rmsnorm.py,sha256=05MiYyNBlFTkq_YtmI9JtpGpkZ_BOsori--5PdAEyp0,8651
transformer_engine/pytorch/ops/fused/__init__.py,sha256=m57Y7MUSFuByjTWV74NfhJZAC2AXm7WsE4q7svlpMpQ,728
transformer_engine/pytorch/ops/fused/backward_linear_add.py,sha256=0qX4svCDSZ1XTuriZnVeFZ4arPLteeWXMqmxOx-8xds,4819
transformer_engine/pytorch/ops/fused/forward_linear_bias_activation.py,sha256=BeeOViIXi7hnTiMvUr_Lv3uj5yKMLtUO-9AtB5We4Sg,6635
transformer_engine/pytorch/ops/fused/forward_linear_bias_add.py,sha256=MFcrRCznVv0afjxOv_NxfgF2piu_GF5ypeMMBxNwJ_4,6303
transformer_engine/pytorch/ops/fused/userbuffers_backward_linear.py,sha256=kwRqZA2KnONumL6OEL-4zkTf2M2nYuTu1COVkHMtW68,30242
transformer_engine/pytorch/ops/fused/userbuffers_forward_linear.py,sha256=1zIFvev3sc1lmF7LZkrOOJh7qoFUxPSHMn7CmnufdvM,22205
transformer_engine/pytorch/optimizers/__init__.py,sha256=Dla_bDtO7EIz8gMkqFJBwqQP5dvf81gXvugKrb0ZJZo,579
transformer_engine/pytorch/optimizers/fused_adam.py,sha256=pAl-zNLcnpsTKvPGR5mxwpaeIhCjk9bNyAFzlQk9P1U,33312
transformer_engine/pytorch/optimizers/fused_sgd.py,sha256=vXFFEBwqS7Hwg8OWt-KbVvFBT_9aVyxPLctbI5VDlZg,11940
transformer_engine/pytorch/optimizers/multi_tensor_apply.py,sha256=t1ISgz6cQc84NFIVFc5qPBCLiU9N2zvZ4PGBEObMhfY,760
transformer_engine/pytorch/tensor/__init__.py,sha256=44KEenmTA3A_FscnYX-WS_93PnPrJIq1g4XNvELQa4Q,2046
transformer_engine/pytorch/tensor/float8_blockwise_tensor.py,sha256=OizAEUTjEUnUf_HSLmjlWmztdlNpUyzIVbNnXYzsP4k,21534
transformer_engine/pytorch/tensor/float8_tensor.py,sha256=3KeExbnpo1lGur9wqCWEstzrfHIVXwiVMcNRe4cHOeA,27449
transformer_engine/pytorch/tensor/mxfp8_tensor.py,sha256=3lz35N0r6Ksk6Y18-Yn9fv9uzG2Kxwp1OXIlT8aYzwg,21042
transformer_engine/pytorch/tensor/quantized_tensor.py,sha256=Sn-aatN4MpZcQFRxS395yNFZeXC8bpVjCyDfE3g_7Ns,15847
transformer_engine/pytorch/tensor/utils.py,sha256=cLNMs96PonjQHZJm60savEch8d5h0uXCH35HVN3zG2E,15021
transformer_engine/pytorch/tensor/_internal/__init__.py,sha256=WT49hCxtio4bYpUN75_wruxHTuDnyqOkZJnsnIKRgRI,176
transformer_engine/pytorch/tensor/_internal/float8_blockwise_tensor_base.py,sha256=qZtbQXbeRIFcJnxB1fH1ljNV-CXCzdR0RhGgDjObI0g,9419
transformer_engine/pytorch/tensor/_internal/float8_tensor_base.py,sha256=mYlN4oBD392v3K3xSLtrUyI2tVh9CZNTyVdctkHrUzw,4860
transformer_engine/pytorch/tensor/_internal/mxfp8_tensor_base.py,sha256=_9ke8oDWeBtQhYk9UWiEaCfVy3I1HNk7PX0yAdZ5ZA4,4557
transformer_engine/pytorch/triton/__init__.py,sha256=CreWKwqjWmIlw_mB-mB_KPrGm1jwWQPJOprXxIdvacw,165
transformer_engine/pytorch/triton/cross_entropy.py,sha256=LIEFoggreAuvYRYPlYjcJGiFw58bB_a1WKHr6m0MtM0,11381
transformer_engine/pytorch/triton/permutation.py,sha256=ndVC_FRVaIakoYL-NP7CA5giJdejDJAWWrek7MDTzxs,23372
transformer_engine-2.3.0.dist-info/licenses/LICENSE,sha256=vqG0Jm8UmfpLuxbW51d-a1tT0e8BCDWp1GCHYCLMhyI,10142
transformer_engine-2.3.0.dist-info/METADATA,sha256=iA5KjZqlLagPSwVgMbOHBqvhmTOcbJCokihTo3joJDQ,19655
transformer_engine-2.3.0.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
transformer_engine-2.3.0.dist-info/top_level.txt,sha256=iWl1kDrBo4b5u2DyflPgpugikjyWbZ-_B3rrbSRoMnQ,19
transformer_engine-2.3.0.dist-info/RECORD,,
