transformer_engine/__init__.py,sha256=-9ZTuHPptovQPv-9a2_ClIfOqbfvk4P8h9wH9vjaq3w,2852
transformer_engine/common/__init__.py,sha256=_89kuuPwEcpxOhljEZS9QtC0jUgMGoYK3PyIDLeRItc,10715
transformer_engine/common/utils.py,sha256=jwsON-BLagT8lPr966PwmapbsUyDMy6BmyFpK0z6mlM,1646
transformer_engine/common/recipe/__init__.py,sha256=CJZJrpu8QriB4Al5s-vLmvdRMfj7aRToVeJqaFWQjv8,14618
transformer_engine/debug/__init__.py,sha256=7Cz0u61rSC58Hp2LVUD0pVoPwaNzAo80J2gN36e-DNQ,309
transformer_engine/debug/features/__init__.py,sha256=yx2oQDw16xc3iYj1rZECaQ0cWJjttmRj2YUtogMDYk4,145
transformer_engine/debug/features/_test_dummy_feature.py,sha256=5MTL3CqDfLNa0J3juiYqM0q9g0Ibpn-Xs6Eu9T8jb60,895
transformer_engine/debug/features/api.py,sha256=Lg868lO0S4mbW7Vp0xkh0Dn1OR5c8NhUyKpW1JVcE6I,17202
transformer_engine/debug/features/disable_fp8_gemm.py,sha256=2qCe9EjhFDzSRqkl5S33DQHxbte-_Ay64agEnoZJKHs,1578
transformer_engine/debug/features/disable_fp8_layer.py,sha256=1BSNbU5QWD57vg36cg4oM5qjOgdwM1xDQSrz_35Irow,1998
transformer_engine/debug/features/fake_quant.py,sha256=-gw-8ZJEtLPZ3QtHF8213fZDX0jwgxyJxswXHmh3V4Y,6144
transformer_engine/debug/features/log_fp8_tensor_stats.py,sha256=iXa-TjZNN7Kodk7cKmqwQ6JWL3a5yTCUpTvBPTdzH3s,6134
transformer_engine/debug/features/log_tensor_stats.py,sha256=5E3WS-euLijdarx1O869rk_H1tigj0GwFVUq7AOYm9A,6248
transformer_engine/debug/features/per_tensor_scaling.py,sha256=5qB44IRXg2V8-dqh7QQDPG2Ti5tSjpGhoL1h9iVYwkw,4290
transformer_engine/debug/features/utils/__init__.py,sha256=DOaTggg-FPhus7L_optePDXHfNodEQVoJ8Vv8xYaiZU,161
transformer_engine/debug/features/utils/stats_buffer.py,sha256=d8VUpcbDYgfnqOYkQ_xphOxVv9ZFCkN3bso4jVENYqU,6867
transformer_engine/debug/features/utils/stats_computation.py,sha256=9dhqLYWXyejpf5xh2795vdjtVjzMMljshBXa0FqU7jY,4743
transformer_engine/debug/pytorch/__init__.py,sha256=reUoLQOU3obVMuEs9PACIOu9ql9pPST1QCvM_C-DzSQ,122
transformer_engine/debug/pytorch/debug_quantization.py,sha256=sEtza0I5VeixbE3ic2BGBQrHgKuAUm5EBubv_LbbSYE,22168
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=N--u19wKl5iAJMNvmDYLdrVGiWMNM_TxdQ2msx3eDdQ,2214
transformer_engine/jax/activation.py,sha256=rlv66ENkpJg4NyrWUwwFYpj6mM9E-tey6Y9n5oNd9lk,2846
transformer_engine/jax/attention.py,sha256=kal0ryqkeXAvO7peaaaI67vFCivfAqIb7iTBQi82vLg,43997
transformer_engine/jax/dense.py,sha256=Lb-JY8x9wYKorBYwhoXFC6PgfXZTwt5f8YtHI6m9IOo,14260
transformer_engine/jax/layernorm.py,sha256=0kyNSbjUfVzgX_ljf74Cffmn-5QJxONNejmnPp9AHWU,4454
transformer_engine/jax/layernorm_dense.py,sha256=k0YwwFcnPCRPT-g_gkpXb9zk82KgHP8pSOMqfZbAHis,9228
transformer_engine/jax/layernorm_mlp.py,sha256=kZYdTRliLhy9WbYfasPrgfzstgwdq4eGu6nFDBEspAo,15831
transformer_engine/jax/setup.py,sha256=kgBIn7h7q8wQAj20BlOrhUYpUkydpUnCUxtgI464jGk,3463
transformer_engine/jax/sharding.py,sha256=99sQg9B2blb5pL70UNVhxjAhNoCqn966b8bNW9JiaLE,11948
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=5s3pQEQRUIra1kg-Bo1ZVCu9wsvf6qc9X-rgMCKVIOs,41832
transformer_engine/jax/cpp_extensions/attention.py,sha256=IxZreEKa96t_7BZwtjPUwiOCvYazLAFfLFqoHhgf_pE,109928
transformer_engine/jax/cpp_extensions/base.py,sha256=HrNNcF5x-FpDLXjft2mcVRZLJT0IT99matPt964uTbA,4252
transformer_engine/jax/cpp_extensions/gemm.py,sha256=g6FjemL3rbWS64eqSB-Eh_F6Qq0z0E9Chf-e-2W5Hlg,21647
transformer_engine/jax/cpp_extensions/misc.py,sha256=Z83lPqOBMchgUlcyLhS0zuBTaBLw03CgwZBBwtMjQb0,9271
transformer_engine/jax/cpp_extensions/normalization.py,sha256=_hJZtsyLlwUYpNJ7rcE0_JWzwiBawshc34_fvExRZMk,46982
transformer_engine/jax/cpp_extensions/quantization.py,sha256=4YGt0V8h_5tVVRmKjJfAxLtOz3Wvw3vHcNFnXvVRrfQ,34520
transformer_engine/jax/cpp_extensions/softmax.py,sha256=9M74GmMwZLf85ArEYK9-NrPDksvy9IThzdmMtZERJV0,30072
transformer_engine/jax/flax/__init__.py,sha256=wohmcM50EsahqVjO2IowXtGhO7IhGOVUS9HmJvoJz7g,737
transformer_engine/jax/flax/module.py,sha256=mSg-Rg0yXqqEh-IiqGK9ZET3quiYVr40Hvb_384FKfk,51903
transformer_engine/jax/flax/transformer.py,sha256=2uEbh-Jnt1tOMu4k8MtehYv8U46d8sTauZfChcpd4d8,93210
transformer_engine/jax/quantize/__init__.py,sha256=jX3FgLCuhwFdCPLXQ2l8kTaQj3LNc237V3b1a6VbKG8,595
transformer_engine/jax/quantize/dequantizer.py,sha256=LDmfZMW9fcX61GqcQEX5kMT31qt7cmYR4oRUHyCYYxk,7490
transformer_engine/jax/quantize/device_utils.py,sha256=qzCn5r2KZlS_mtoLb5dCU7XUNHPJZvJCtQSJKb71wWs,945
transformer_engine/jax/quantize/helper.py,sha256=zuLqdLmBsUUAohtBI-GMcDIwO9zCin5FevLBwS6A6pY,16273
transformer_engine/jax/quantize/metadata.py,sha256=mUcgaxEldPjekCuEEJr1Qjn1du-DudwoHoosQvZJxu0,1038
transformer_engine/jax/quantize/quantizer.py,sha256=34W7M6S5U227CiUlpmFg6J-6Ax2YHBwCuNYhQJ-nSJc,32983
transformer_engine/jax/quantize/scaling_modes.py,sha256=YNkUpqrJTDLVwxK5Iu_RD7WfTW-3gow3IOaAbDQP_WA,26296
transformer_engine/jax/quantize/tensor.py,sha256=Z9qA7buuDvN5-ZXWcmykWEfwM4HQlv1ctKqvraB02vI,22613
transformer_engine/pytorch/__init__.py,sha256=CbgZY_mhmcC8JogCNLEoMAtEaR9N485HuOk8PnR5O_c,2433
transformer_engine/pytorch/constants.py,sha256=ZtT-1WhUWmNBLxD2P2rtZBUAE7E-MdaVY_uZFXgGPtc,2091
transformer_engine/pytorch/cpu_offload.py,sha256=pekfMeFWMGlCp6Po2hSzXkiq_lZMfnYCzd8BG32v2vM,28676
transformer_engine/pytorch/cross_entropy.py,sha256=kR7qaPp4Rfzs_F-0DwHxglmzLa6ipv4VQLg7UW03oUM,2871
transformer_engine/pytorch/distributed.py,sha256=-1TBzRLeFomEzd2t8iFlA1ce3u3ITcITKKGqE1w7zuQ,65118
transformer_engine/pytorch/float8_tensor.py,sha256=KdHMbjsSpPWkB1Cu4d0Z01QhATMijK03e_XCl1p88fo,232
transformer_engine/pytorch/fp8.py,sha256=CZlRxYD2klyBQ1Wosz4GSyhpjbEDhdnlGSXyIExHheY,39621
transformer_engine/pytorch/graph.py,sha256=q3gfJvGa7JJypufyMlVVZKBC1-MCIyEGYtXQcYFQBrY,33869
transformer_engine/pytorch/jit.py,sha256=Fo8lLdj2DFeRv0tvcuJDZD73yXclvUfPUo_UCx_-kpk,12635
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=ToQtLwsz4u1GkTVwZOoL6NoesN77GG888-KuXkcpXW8,2063
transformer_engine/pytorch/transformer.py,sha256=GnYMMw44QdedrXCPPWe1V6Duck0RHeK82PziKE2gAHo,42046
transformer_engine/pytorch/utils.py,sha256=vZh1dUJ_GCH41S4I-K11CC61wxwyHGy9Jp6oY2E7InE,21617
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=rdSZqC4aeR_dJ4g5TOo8_UuiH7lsqC9dnYQtu0rQSuQ,43946
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=w2_FuOMtsvih49tUQgwT-jrmZGOhDwxbWYSBVdKpgQc,70601
transformer_engine/pytorch/attention/dot_product_attention/context_parallel.py,sha256=Qh39ijX7GiwwlbGQBCfxTkTj3MMkdisr1j1ZtbWRDHI,184917
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=UK7BpHxY42NzEmTtzdAo6tFNo6bhqpWQZYz6UH0neMQ,78857
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=wOvPalBgG58wkT7YyUIQOQ89M052bJe6-9_NyNKmtxY,5989
transformer_engine/pytorch/module/__init__.py,sha256=JHc9_LIv9Uoe4N-vyiGxS5GspAu9WWl4fNWyVQP8mAE,506
transformer_engine/pytorch/module/_common.py,sha256=NckzdswAv3Fo9QEqTNeDmoXKdGVrzzlx91eljNlcyu8,8354
transformer_engine/pytorch/module/base.py,sha256=KiI0ek0u8-fhb5XfJegy-ifLe3xbfjGOcgbcuC6wbsE,64210
transformer_engine/pytorch/module/fp8_padding.py,sha256=ivCbRi2lF-vbnKsMDuzy6579rZZ8Wdh8SFVrBqMOdGE,3874
transformer_engine/pytorch/module/fp8_unpadding.py,sha256=X5qfYLy4DFeDbZAjHCiYz4HsXp_JXKQO7D6UYzdnOSE,3876
transformer_engine/pytorch/module/grouped_linear.py,sha256=bigaNAalkLlBuSS0qrM_8OAQkS61-lzr7Tp7s8cV7Fg,35421
transformer_engine/pytorch/module/layernorm.py,sha256=DY8PcT7wf4VksG7H8_9QebfVQL-I33TcGfEJmgqJU5U,6360
transformer_engine/pytorch/module/layernorm_linear.py,sha256=u-QcB-4Js7PyjrefuPEcALeTp4GMamkcwXt0CdEta_Y,74657
transformer_engine/pytorch/module/layernorm_mlp.py,sha256=Timskj8OVO77S1d9OK5cDyMpAJT19tINmAWCt8jPR5E,92087
transformer_engine/pytorch/module/linear.py,sha256=hEFIAJI7rAmxHJ4r7nOJjEpHFofsS_RFkMbKWkHIwVc,66301
transformer_engine/pytorch/module/rmsnorm.py,sha256=Twbov8LWcdvbYYltS_DmDf4QVNaeYeB39eIafRHLTQ4,6406
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=U0itmbrXyCE2GPC0ugCVsMG9HOYVFQoRtD3HiGN-xzc,14474
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=FbHK2JBaPo0352os_9qMHEqB68ggkOvLOWVq8cx9cjY,700
transformer_engine/pytorch/ops/basic/activation.py,sha256=GiWgmWrtXLAiU4dahjbHckb4CsL1gJPVcZfowNErk3w,9743
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=9MkXzY-1HOe39gjEI0PImRbK77zsLIWy1hISHCD1RWY,40576
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/l2normalization.py,sha256=qMabObVR79R5EV3vevZcbVOZWKBrRX6sRIxKWBpBWlA,4507
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=juI3LS6wDi2BLORZ3Q88CmQlJiYLsAhTX4eVVI0qyx8,4806
transformer_engine/pytorch/ops/fused/forward_linear_bias_activation.py,sha256=P7Z39RHQUc9QXtUbH2jEXld0jzyIpqwX_vnc-B8F5CI,6954
transformer_engine/pytorch/ops/fused/forward_linear_bias_add.py,sha256=4HrLjXsXuV2D4AqtEl0VqsFgFLnhNbHsDzkP8QPMIOI,6622
transformer_engine/pytorch/ops/fused/userbuffers_backward_linear.py,sha256=oIH1lyhUHZ7sqireai2anTJ3aZApuUh7FRM244M7Uu4,25658
transformer_engine/pytorch/ops/fused/userbuffers_forward_linear.py,sha256=T2MJEvkH1BfCREO5piISowZb76CHwa8s23DvWD67LYQ,18466
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=VDgsCbtuarkcZlqmPwVngaTbZ3VEGjlWZ-N4mol_Hik,29894
transformer_engine/pytorch/tensor/float8_tensor.py,sha256=aGiovI2HYXHPgX-T5S-jUoIDB_c7XqN3_h2pGasD-WE,25911
transformer_engine/pytorch/tensor/mxfp8_tensor.py,sha256=4RQWpNdNzpcvyWG0Lgx_zzhzpMbZiNl0egx0e0Z-Wcg,19268
transformer_engine/pytorch/tensor/quantized_tensor.py,sha256=lEX072Opc23sLkM_HgssrSpXrD1allzioi6h-CxUx9M,18557
transformer_engine/pytorch/tensor/utils.py,sha256=WRa03Bm4RdzJ2hy8cI3uvdgzWsnPaK2_3qV0w4zUokE,21697
transformer_engine/pytorch/tensor/_internal/__init__.py,sha256=WT49hCxtio4bYpUN75_wruxHTuDnyqOkZJnsnIKRgRI,176
transformer_engine/pytorch/tensor/_internal/float8_blockwise_tensor_base.py,sha256=KssBJ1QeoJ-Fc1Oz_UzGV8NCFTqnb1md3OMyUR3APWc,16743
transformer_engine/pytorch/tensor/_internal/float8_tensor_base.py,sha256=C7ORSD2adlLd0FJcjLEcrYL91qbD2tQAj3GF3qffa8I,7027
transformer_engine/pytorch/tensor/_internal/mxfp8_tensor_base.py,sha256=vZcdXUIHTeej0fxjC7VeILltSaxh5Db2Lnb3sywQZdI,7080
transformer_engine/pytorch/triton/__init__.py,sha256=CreWKwqjWmIlw_mB-mB_KPrGm1jwWQPJOprXxIdvacw,165
transformer_engine/pytorch/triton/cross_entropy.py,sha256=CXAmXPq8KqtCEf5E2DnnOgslgcPPnPsqop69Neo1LmQ,12345
transformer_engine/pytorch/triton/permutation.py,sha256=ndVC_FRVaIakoYL-NP7CA5giJdejDJAWWrek7MDTzxs,23372
transformer_engine-2.5.0.dist-info/licenses/LICENSE,sha256=vqG0Jm8UmfpLuxbW51d-a1tT0e8BCDWp1GCHYCLMhyI,10142
transformer_engine-2.5.0.dist-info/METADATA,sha256=w8ubDPt2fwXAiBTlEUVhcPtNr8mFkyn7BRlSpeEAygo,23327
transformer_engine-2.5.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
transformer_engine-2.5.0.dist-info/top_level.txt,sha256=iWl1kDrBo4b5u2DyflPgpugikjyWbZ-_B3rrbSRoMnQ,19
transformer_engine-2.5.0.dist-info/RECORD,,
