transformer_engine/__init__.py,sha256=B_2n4nkVvyvZwMrg0BPmkytofWBm80OYNXWN3paXTGg,464
transformer_engine/common/__init__.py,sha256=wuRE8GSCFfahrW_JgVrPGL_Fs8lugcegndnx1nCvvcs,12450
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=ctccOnaTBD8K1LDTFKxpljuoYiUrqLLHi3IeYFDI-1A,6664
transformer_engine/debug/features/utils/stats_computation.py,sha256=QFUPHrUK6-voQy_RZhyzJkWco7UniX1qhnA1trEanJc,4670
transformer_engine/debug/pytorch/__init__.py,sha256=reUoLQOU3obVMuEs9PACIOu9ql9pPST1QCvM_C-DzSQ,122
transformer_engine/debug/pytorch/debug_quantization.py,sha256=ibZNWqQXV1CF1-if-ND0k1K3z_r1F9LJhpFwLY_JLFM,21929
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=OTNm1oiTNeLwhdptQ6byRBzsQ1U1PrWKihPte9RjHmM,43908
transformer_engine/jax/dense.py,sha256=rYjq6H9DYWiKyCvwZeMZ3e7PNh7hfX5ifwdxdkntvi0,10969
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=kroHlxwDtxXi_jj3Dy4nVcNmZMRmUwmPB6yvpP2O9wU,15746
transformer_engine/jax/setup.py,sha256=9XbDM4Bvp0SQgR29cN5Xaxjy9Llh9CfDG49hSo8IUGY,3804
transformer_engine/jax/sharding.py,sha256=ujiloZocFGofpakzRLsSC3h_IYJJ093fiqbKNgQnIZ8,11041
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=S-FbUdhueIsVfc8JTsvmSMeOoC7jieqFpluxd3aYgyI,40981
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=ZWPD4snw4_WpGuRUxtSWluMugQHgtAOZDR3rgbZdZws,18396
transformer_engine/jax/cpp_extensions/misc.py,sha256=MUqr6fZ5w4IfmzALZ7tXWqTbr4G_oBB2qEi9cmZNn_E,8990
transformer_engine/jax/cpp_extensions/normalization.py,sha256=_hJZtsyLlwUYpNJ7rcE0_JWzwiBawshc34_fvExRZMk,46982
transformer_engine/jax/cpp_extensions/quantization.py,sha256=MuneDa0q5bX4GvJuUIDPNw_YCoIrJFLGvoxQAmLSYB8,22603
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=mSg-Rg0yXqqEh-IiqGK9ZET3quiYVr40Hvb_384FKfk,51903
transformer_engine/jax/flax/transformer.py,sha256=pEgc_LTZi3gEo0wd8XNVrV38lzGuMi7Y3gf-dzYPBLA,92821
transformer_engine/jax/quantize/__init__.py,sha256=j-g3rSZF211AhI5T26jnaAKEIGbt0791amtU3DR5T2o,567
transformer_engine/jax/quantize/dequantizer.py,sha256=tmPp06noEqvab3ETMuJFNTdzLyVJf3amISWudT7JjB0,3948
transformer_engine/jax/quantize/helper.py,sha256=CtcW2y1hei3nhRgNxB7paYwr5g_7KsS_SKcfyIAjGe0,16353
transformer_engine/jax/quantize/metadata.py,sha256=mUcgaxEldPjekCuEEJr1Qjn1du-DudwoHoosQvZJxu0,1038
transformer_engine/jax/quantize/quantizer.py,sha256=ZfFvSwa5scA5kCXMJXbBgmF1oUoJImSwbPomA-fAImM,25144
transformer_engine/jax/quantize/scaling_modes.py,sha256=Sfs9wwoKABlebqoC_yfdlYA6h9kw4KDx-Jq4pdQ1te4,16699
transformer_engine/jax/quantize/tensor.py,sha256=2gqTaVbz6cnX6AOEToHYaYJMkdISHs8BW3hwyEIMlHA,15610
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=SY-BdDZZLrJ0iHvkJY-OiNIjpKTAX-l6wU00vTcRTvc,25305
transformer_engine/pytorch/cross_entropy.py,sha256=kR7qaPp4Rfzs_F-0DwHxglmzLa6ipv4VQLg7UW03oUM,2871
transformer_engine/pytorch/distributed.py,sha256=M6QoGvs4glC4RKGJTpVL93ICnF4DzyZfiVbJgedvQvQ,59295
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=ROnVpBpzsn-wSX3ps4nlGxvJKOC_EtuT8UEQcnmtGNw,9164
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=GlBimokL3ontMk1JkKElP0C6o7czw3Px0Rzj4hWvZWY,2332
transformer_engine/pytorch/transformer.py,sha256=Hksjs327wF70SjaykJOJ9NfSnypvq8-J7cdQ6RzdUP8,40244
transformer_engine/pytorch/utils.py,sha256=7lXk8SZ_x7EcuPLYP6wtH_7WBdvcX4elEsf89iSB_mM,21283
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=zZecaTg17w9SghOWuZ2mXHWMZuUPIRrikkqtcJdivS8,40697
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=su197EcBWS8a-vaofcAz2L3L1OP5bENPkQsfgc28Amk,168951
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=FyxqnsMgHkdmvh8re1VHoFtCR6jqC6bvf3e7s1MXWD8,78135
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=59WhWgU2RFXnW3ukMP1TzI2Y7SxLvxA-7maEp965pNs,5672
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=nSHCdKp-pdxj-ez42fzWjqmcM2hCGhl_km4wZm-Tl6E,62346
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=o7id2quOuypKJShEkbxNtw-CbW-20gMRVUwew6h-P3I,34420
transformer_engine/pytorch/module/layernorm.py,sha256=DY8PcT7wf4VksG7H8_9QebfVQL-I33TcGfEJmgqJU5U,6360
transformer_engine/pytorch/module/layernorm_linear.py,sha256=apQvVIW_et1ZfSx8X2bkIherOK1fLR4ZDDOFe3oywzY,73059
transformer_engine/pytorch/module/layernorm_mlp.py,sha256=WQNC1OJA71Y46-dKsqRsEomW75OPq_j6Jms5P680s-A,90491
transformer_engine/pytorch/module/linear.py,sha256=6PJ33x-ZaHrD0zGkY9Ss1JxHoTERqzZTb7EpsjN4pL0,64899
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=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=B9u1Tp5jk6PBGBmbvfT6eA_ytk4QZAoOMCfX_CM3mEM,37661
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=Mq1a0XyX9ZteEMvZvH3jicjPsRCz1lhOb-o2jOUjifk,25001
transformer_engine/pytorch/ops/fused/userbuffers_forward_linear.py,sha256=ZEa4HQhWCGqwIPlG4nnSV3_-bVexiejUgZmm7YzpzAE,17090
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=j4dJn-t14WIHACAU1k-dvTQ8Axh_ALOLlaxprVVZ0i0,25525
transformer_engine/pytorch/tensor/float8_tensor.py,sha256=aGiovI2HYXHPgX-T5S-jUoIDB_c7XqN3_h2pGasD-WE,25911
transformer_engine/pytorch/tensor/mxfp8_tensor.py,sha256=XZokchgIIRoDOOAibJkBQoslU5jJPOyLgSHotPxvZm0,19227
transformer_engine/pytorch/tensor/quantized_tensor.py,sha256=N9LFlf5oO8bEEEkl6LGlzs1aT6vqwjUebZKDSZgaAbc,18123
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=Jp0Fg2Tz67kXml9d0-cCMD9fVbQg9svDIfngc1XIC5c,13851
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=B9rj4aI4KrUmnW3RYbegnlkomitqLAbo60DBbyISB3c,11766
transformer_engine/pytorch/triton/permutation.py,sha256=ndVC_FRVaIakoYL-NP7CA5giJdejDJAWWrek7MDTzxs,23372
transformer_engine-2.4.0.dist-info/licenses/LICENSE,sha256=vqG0Jm8UmfpLuxbW51d-a1tT0e8BCDWp1GCHYCLMhyI,10142
transformer_engine-2.4.0.dist-info/METADATA,sha256=AT2mkbqYaPMS8eZo73y-17Vl2uxnXbsdO9t9HOZOKM8,23239
transformer_engine-2.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
transformer_engine-2.4.0.dist-info/top_level.txt,sha256=iWl1kDrBo4b5u2DyflPgpugikjyWbZ-_B3rrbSRoMnQ,19
transformer_engine-2.4.0.dist-info/RECORD,,
