transformer_engine/__init__.py,sha256=8YzGGwTUlY1dAb1mbq0_-Ob8X6naQeBHmAB9bJP5c2k,535
transformer_engine/common/__init__.py,sha256=Zf4mWYDW9n7PFigdRI24YnDwoXaVZT4BZ9jLY_UIZ20,5867
transformer_engine/common/utils.py,sha256=jwsON-BLagT8lPr966PwmapbsUyDMy6BmyFpK0z6mlM,1646
transformer_engine/common/recipe/__init__.py,sha256=3zAJaX3-LmXhIVOmlxKZYYMXQ0MLx5DmxTp_AkH98T8,8398
transformer_engine/jax/__init__.py,sha256=Z-0X7hh-o15S7cyd-VIoofJNMznou8kqiyU24FCEkaY,3668
transformer_engine/jax/attention.py,sha256=mAFkT3kOfjhj98kzJPKIfJe5PxfDLG3UCIqu2SJyj9Y,38770
transformer_engine/jax/dot.py,sha256=SAILtHK7X_86jg8Af7VFCmJsZACVG3CNhydp-7-v74A,7290
transformer_engine/jax/fp8.py,sha256=gNwRAouZbLIpoXFu4e_ox3HCQJtRAE6Jt4kLakKYVq0,14968
transformer_engine/jax/layernorm.py,sha256=oBydG8Tbomr1RVPoIYFm-_VWQ7vfcT9Ny9X8g3DPmYQ,11207
transformer_engine/jax/layernorm_mlp.py,sha256=m4d-63wRjy-_YJ8H0jPaGpJJrNiZki6hwOANh9X8bm4,19464
transformer_engine/jax/setup.py,sha256=izrAQ8TsSfOXGpciqMZKqc3OzkOpOMNiHgT6aNVrDYM,2578
transformer_engine/jax/sharding.py,sha256=zm2NYFff5aGcwVrFld8mpMVUHOMrDUvce-QuENE-9Ng,8696
transformer_engine/jax/softmax.py,sha256=IujIGDvRzVJQ9bm3lH9PqwuM77aOljcfkNtZ7Wdf3xk,2051
transformer_engine/jax/util.py,sha256=YaY7qZOWgQ-cKPCm2UlsKyjE9GEhL-XtMBIzP8UAZto,815
transformer_engine/jax/cpp_extensions/__init__.py,sha256=8Orci3CVhULl8krO6HqqwE532D7vpttUGjS05CwBiw4,320
transformer_engine/jax/cpp_extensions/activation.py,sha256=pxGLkALk7dsEbT61kJDYHkODUaQGEdJcbgxv_JEicqg,17885
transformer_engine/jax/cpp_extensions/attention.py,sha256=aX7dbycTz4HKCNDL0CPir9njjgPUya6XOhDvK9Oxzc4,95214
transformer_engine/jax/cpp_extensions/base.py,sha256=iKjZGweOcFZj5YQSi2kkEUTMjeMOjJtxHUP9nL9qKsY,3658
transformer_engine/jax/cpp_extensions/custom_call.py,sha256=5nrng_6OGlrJg3XdGGca_kt8GOEZW2oqzuFvoB-zRZA,3737
transformer_engine/jax/cpp_extensions/misc.py,sha256=EQ7e1dmPwqNoBDxT_MkcCARfrGrwzD56J0fH0xEYqEI,6461
transformer_engine/jax/cpp_extensions/normalization.py,sha256=Dl2LuDyIefFvlU0G5MzDdv5SkMW4_h545kJHIeEgCps,56899
transformer_engine/jax/cpp_extensions/quantization.py,sha256=qjKl-2m9CxN-PyhXCtETPkYcypcuZqf8ycaJpyL92FI,6872
transformer_engine/jax/cpp_extensions/softmax.py,sha256=JBdCWoB6J5bu0-xpSiE5JRj2yfFz4tK0uAHrYK7hF4c,33247
transformer_engine/jax/cpp_extensions/transpose.py,sha256=R1HkOHdR5VdQGK5WLMdlFkuSUe80gAeMYiBVynOz_MM,46300
transformer_engine/jax/flax/__init__.py,sha256=rbxeqz7kjcC8Q_0MEQoWr05wLxgfFDROx3ZH8C3xDGk,789
transformer_engine/jax/flax/module.py,sha256=OCDbi7hU8LxxWGC2UYZv-zAPEA3aGnMi1mFdsodfKE8,52854
transformer_engine/jax/flax/transformer.py,sha256=cNX5fjmDYrSqPKkt5w89kutjDf3X0XiMhvYibzdiAtI,92399
transformer_engine/jax/praxis/__init__.py,sha256=3NqgRc9TGfar3-7z947rHyWApy4OcqxVZiUmH1TikUA,464
transformer_engine/jax/praxis/module.py,sha256=dqwPxVZZ1N6noY4pZuQKrGTdc5ZdNo7OzgaZ9FV8Nj0,11331
transformer_engine/jax/praxis/transformer.py,sha256=PXZPLVpeVz4-01JLLdzBVrRzGtt3MOuKs0NrjkP-eOw,16365
transformer_engine/pytorch/__init__.py,sha256=HLCWofMK8mPUyNkUhjXEF-e7Jj0BK62O0Lj7YUhg0LY,4393
transformer_engine/pytorch/attention.py,sha256=95x9CggPvwrRyEScgk9cEwSuwHU8aye9HPwXTxTZsos,394225
transformer_engine/pytorch/constants.py,sha256=2n8n__34wvCeOWmjh5XbwWXGud3757tIEj6RVpjhZyw,1703
transformer_engine/pytorch/cpu_offload.py,sha256=qt3TrxJqrmAvUwOhqchQjPOlqWqx03_f6drA82Vz3E8,20780
transformer_engine/pytorch/distributed.py,sha256=UEQr3yxZeyk3W7DCm-Dn9uCupSV7YxO94SFhQNuhqMs,44541
transformer_engine/pytorch/float8_tensor.py,sha256=KdHMbjsSpPWkB1Cu4d0Z01QhATMijK03e_XCl1p88fo,232
transformer_engine/pytorch/fp8.py,sha256=TZRl0U0gal802JGD3__R-2u5CmpCZ05rDT9qDUJJ2gY,31042
transformer_engine/pytorch/graph.py,sha256=vPCLSITHkDiGEylMER8LMOMsVBJ7ZyQlBgta6wHvxrA,32984
transformer_engine/pytorch/jit.py,sha256=KWnIyCq53vaFbYqzYdhfTufSJpHEGcKCU45aTauJy0E,9305
transformer_engine/pytorch/numerics_debug.py,sha256=dayXGDf0_qDtPBhRQQxMIdsSFzdkRmaeUF0PVNVmGaQ,908
transformer_engine/pytorch/permutation.py,sha256=N-mGOdbTPq0vXR3WpEynDw_b6KDshknga4t6-HqvXQ4,26939
transformer_engine/pytorch/setup.py,sha256=0OheemxVdkYO1v_zlyBwZCCzKiTIUeYDje5ezduFu7w,2379
transformer_engine/pytorch/softmax.py,sha256=m-xaEG89MZKmejE2c4KVFiXO4NnrAhCOJ5-MetyHyQw,10244
transformer_engine/pytorch/transformer.py,sha256=-VwzTZklrSqbzSWD4YkgSw_y6m9Sa6TIgOE1-oncFmw,39324
transformer_engine/pytorch/utils.py,sha256=FCoZT6nuN-vuXKQWPpraJ9tzaWf8tdb_mXgVyMcNIno,13920
transformer_engine/pytorch/cpp_extensions/__init__.py,sha256=McGPd8B4JzNtIKvaYTaznD9gDTufNYhBozFSRwKnmxs,461
transformer_engine/pytorch/cpp_extensions/fused_attn.py,sha256=Re3T-Jiv4xwxdq5rMOLJFXPo-LDeWajV9SlWjbNGIfk,19345
transformer_engine/pytorch/cpp_extensions/gemm.py,sha256=n8Sz4rFDy2tSM5nA1YKrKKPBU6r02ZXsbxtoSEoGWag,6867
transformer_engine/pytorch/module/__init__.py,sha256=JHc9_LIv9Uoe4N-vyiGxS5GspAu9WWl4fNWyVQP8mAE,506
transformer_engine/pytorch/module/_common.py,sha256=AXqh8YyxNeuZWxaI1lvsLtOTcIYRqLS6QDbfGUB9H3E,8677
transformer_engine/pytorch/module/base.py,sha256=UKWs22wEeZgV5A_v75uwiBU7rA5Yu76aDYOhXkxAFXc,46855
transformer_engine/pytorch/module/fp8_padding.py,sha256=HatfOFJyltQzxAr03MgxpQtXS00jVntAsKkl3KNog0k,3155
transformer_engine/pytorch/module/fp8_unpadding.py,sha256=PH92dzpg5iyS-DRzFV7fo-zYuv93ymSiPV4BsIzgVyI,3135
transformer_engine/pytorch/module/grouped_linear.py,sha256=r9QXrh-h9lVsslLVOustWBybzw9m49loP7TnaiJi-6U,26688
transformer_engine/pytorch/module/layernorm.py,sha256=E11uH1YpKsR78YneQ5LtWVbHD95E0Ps7S4KKvqQ8lT4,6542
transformer_engine/pytorch/module/layernorm_linear.py,sha256=QQfaqSOcMab9BRLKgZB13rmKXmDckDFt_Hs2Wk9aoxI,56770
transformer_engine/pytorch/module/layernorm_mlp.py,sha256=PPaUuGEKTZfshlFV0NF0eyR_yTFQbnxXXktoTMd6Emc,65473
transformer_engine/pytorch/module/linear.py,sha256=0xz9a6tU12OMAlk16B1JqOLiGp57hRgVhFV7FWo7iNY,46820
transformer_engine/pytorch/module/rmsnorm.py,sha256=DaI-M4qkz9KwbXoLAo034i_XTig96fSV_pwr-MGvrWA,6651
transformer_engine/pytorch/ops/__init__.py,sha256=3TCOnORonjWt7x1eW7pg2S2ZRoYC4Q_DGPWkArRCpko,453
transformer_engine/pytorch/ops/_common.py,sha256=YM-ygXKwE5LvC0vM5cJzf52ASimdm5dy6eCyPFZ0Au8,4207
transformer_engine/pytorch/ops/fuser.py,sha256=rfzm7bP2dmWriH1NSbp6jxdvC3a33k8VxasxsgzdIcY,15570
transformer_engine/pytorch/ops/linear.py,sha256=EElBJ20M9ncn5LB5vjOXzWDQm6aJKPDyvEAncUkJUiY,5882
transformer_engine/pytorch/ops/op.py,sha256=wtd_JJxDGa-jFcaoAAfYSaEvAV47AaUTg_Rwkag8sL8,24755
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=xDqTOuxtB7OtzYNa3emvKRaTHhfv-3TbYJ_WNN5i-To,8208
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=r-GKAuJJm0x_C-E4oCG9klJkZQQNqZSPVrTH_azEyvM,37007
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=KZFpbPqqE_ss_UjyKqcv-Sp-BXD3lCN5vF0rn1Pp2ko,10177
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=a_ZlK6ZdEMWUxFCkAhmb3ZuPLZLNeuoVZUfzsecl_tQ,9228
transformer_engine/pytorch/ops/fused/__init__.py,sha256=3-95xZR0tywj3lNtiUQ4X8uwZdTPQ2R8eEK1xZuayjo,963
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=9MJrks0t2nWv8z45IB7OspESvaSBS6bpvGgQmAWLH0s,33273
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=Any0Lr3vsMm1wCphmYHL2DOHS8qECdqI_HrJyqOFUwY,1273
transformer_engine/pytorch/tensor/float8_tensor.py,sha256=H9Md4b2M0ElCPWxnC3cecBvGY8OOQQHUjy1Y1VSJFQU,21715
transformer_engine/pytorch/tensor/mxfp8_tensor.py,sha256=Lqt5BVBzXXCqvWTPb0baPR49BSAiifZMybg2EbVq38E,20438
transformer_engine/pytorch/tensor/quantized_tensor.py,sha256=Z8pKzBYgZe5A_GLo5wyrhIn3YtS8gD4SmirlxFonMtY,15656
transformer_engine/pytorch/tensor/_internal/__init__.py,sha256=WT49hCxtio4bYpUN75_wruxHTuDnyqOkZJnsnIKRgRI,176
transformer_engine/pytorch/tensor/_internal/float8_tensor_base.py,sha256=iYgE-FJFE8s5zBwAD_7I6ILHxdv749H_4L-YYDL91w8,4369
transformer_engine/pytorch/tensor/_internal/mxfp8_tensor_base.py,sha256=ofwO2GOFX3LeL-gIJbazFo6tJbPYI9Md-vHOmrLRTJo,4542
transformer_engine/pytorch/triton/__init__.py,sha256=CreWKwqjWmIlw_mB-mB_KPrGm1jwWQPJOprXxIdvacw,165
transformer_engine/pytorch/triton/permutation.py,sha256=GwQd0_Xym2LRwCICiy1v_IbGIic815Qz0G1Wh2KKkvg,23746
transformer_engine/pytorch/triton_kernels/__init__.py,sha256=EmM4yk8iLlwpB0q_9RgQno_XP6pZOYitifobOuqkzZE,172
transformer_engine/pytorch/triton_kernels/cast_transpose.py,sha256=tYm5JZhGnJZ5NHK8ivWYpLG6eKG855MEqLYiBzpCWio,8209
transformer_engine/pytorch/triton_kernels/common.py,sha256=O5R43rg3x_S9oXvNs4eF_wEXTGxAQSAsUJo5F3-7oqs,3249
transformer_engine/pytorch/triton_kernels/layernorm.py,sha256=lQBa31hWluW592KIELRoW7Vd6ZuC-HfcL3wun_EqFUY,18984
transformer_engine/pytorch/triton_kernels/norm_common.py,sha256=A1SmufDo9yHFOQ_AZmR-5bUHPywlYSLuKJQysJ950-I,1262
transformer_engine/pytorch/triton_kernels/rmsnorm.py,sha256=5r3blvioAROxdyATMLJ5dM756X8NbHGccGTOQL5ai0I,13957
transformer_engine-2.1.0.dist-info/licenses/LICENSE,sha256=BpqnbtLdQYeQ0_67AUnB1_w3IXtVcqS8hQFBXMxh17A,11327
transformer_engine-2.1.0.dist-info/METADATA,sha256=TJ1HfX9kpX4GUzxuHojJE1VIJw2IN0BkfiRrBMYA2_I,29655
transformer_engine-2.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
transformer_engine-2.1.0.dist-info/top_level.txt,sha256=iWl1kDrBo4b5u2DyflPgpugikjyWbZ-_B3rrbSRoMnQ,19
transformer_engine-2.1.0.dist-info/RECORD,,
