transformer_engine/__init__.py,sha256=8YzGGwTUlY1dAb1mbq0_-Ob8X6naQeBHmAB9bJP5c2k,535
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=mXsBaw_7H5h40AV6O60EmgEmNSMtFPFsNEK8pp0s7wk,12319
transformer_engine/jax/__init__.py,sha256=D_aGzGC6QujNbmd2NsB0tenIWKk5HjZGyFDj8ihq1yU,3406
transformer_engine/jax/attention.py,sha256=OTNm1oiTNeLwhdptQ6byRBzsQ1U1PrWKihPte9RjHmM,43908
transformer_engine/jax/dot.py,sha256=SAILtHK7X_86jg8Af7VFCmJsZACVG3CNhydp-7-v74A,7290
transformer_engine/jax/fp8.py,sha256=U-6PX2J0LSIg6BeYW_b7ByqZqYqT_Gy3YsvP3wRVNpI,14254
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=4HHe29OcWxwGrNkc72OAdkTkZiA-c1giZ7E5T0tzMVY,2105
transformer_engine/jax/sharding.py,sha256=zm2NYFff5aGcwVrFld8mpMVUHOMrDUvce-QuENE-9Ng,8696
transformer_engine/jax/softmax.py,sha256=IujIGDvRzVJQ9bm3lH9PqwuM77aOljcfkNtZ7Wdf3xk,2051
transformer_engine/jax/cpp_extensions/__init__.py,sha256=8Orci3CVhULl8krO6HqqwE532D7vpttUGjS05CwBiw4,320
transformer_engine/jax/cpp_extensions/activation.py,sha256=nS-5-KldVjTdn65LCNKsbqwLM-Bovh3_dGjpAxsjFzc,17759
transformer_engine/jax/cpp_extensions/attention.py,sha256=PQaaU9QexGO7EmFiHhAXiiaLv_QGOxQO0ZYJ2tD4vnM,104848
transformer_engine/jax/cpp_extensions/base.py,sha256=4fR61FZC7vhEHYDfxBmcLKZrqqefEAoSRrslT9C-6zM,3460
transformer_engine/jax/cpp_extensions/custom_call.py,sha256=Q3dyUONPb88NAcMTm70WjU6WHdPvVqRBnJHicuKDNiY,3582
transformer_engine/jax/cpp_extensions/misc.py,sha256=tPehlTbxK4EXunQPMO6T9uQlWEZ1lbPB0CNmM1opvJc,6033
transformer_engine/jax/cpp_extensions/normalization.py,sha256=AH6mJK3weFd-wu6A2hiapzJzzmPSPmHJ6Sfte5txnPY,56764
transformer_engine/jax/cpp_extensions/quantization.py,sha256=wyuxF5R3S_vLDrQZ-6VA5Wpft49nje-AbUW3Bmb1We8,6790
transformer_engine/jax/cpp_extensions/softmax.py,sha256=wBHxN8Hib_u0nnFiXeTYF06XgtO1JRmO0QnAkWN1gNQ,33165
transformer_engine/jax/cpp_extensions/transpose.py,sha256=gBtv4lPE2yUijunm93ZnpXom67xOgqbFkg2Cv_lm6nY,46132
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=idjQBt4spr2lHbYv-47Q1k2U4fq7YoLLwd2vOv3I__0,92271
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=6_pRMaYnin7xnyI6TzCitTFN6C8Il359iKXe5Qj2JRo,4547
transformer_engine/pytorch/attention.py,sha256=ETFJ4UubmymAVTm3fOAy243eOoGC1pgP26CxJgRx6f0,334178
transformer_engine/pytorch/constants.py,sha256=5ZcpGNe3ZIryzsY-KbJhTlwtN3dJSaRl750_8b0BRF0,1942
transformer_engine/pytorch/cpu_offload.py,sha256=9agG_70VV3vzh0pw2XTmzA5BjoHTK4t2nnP81DwmUUs,24560
transformer_engine/pytorch/cross_entropy.py,sha256=DC2FBVxWKJiJ0-rTzUSSldO5P4ASPvi7pGLn2OOCycw,2711
transformer_engine/pytorch/distributed.py,sha256=83uNdLDE_jeWMnu_teSel6igSTiVrZCftjxcwPmh2fc,47803
transformer_engine/pytorch/float8_tensor.py,sha256=KdHMbjsSpPWkB1Cu4d0Z01QhATMijK03e_XCl1p88fo,232
transformer_engine/pytorch/fp8.py,sha256=-kebIYYAXpHtKviLEcGabaojzb8u_nHux41bsuMYO3Y,33679
transformer_engine/pytorch/graph.py,sha256=FxzGwNVRe2TaWSh4Z0EKlP13jdVQd-JyY-3613e--6g,33774
transformer_engine/pytorch/jit.py,sha256=aTxyCHUOOj9cExb9utiQVibjPFnYhBwWQe7Bqf9FVMc,8745
transformer_engine/pytorch/numerics_debug.py,sha256=dayXGDf0_qDtPBhRQQxMIdsSFzdkRmaeUF0PVNVmGaQ,908
transformer_engine/pytorch/permutation.py,sha256=jneuynEvd-DCUqmBTHOtPqgmJXRm9RxASQwxJgLJCkM,26816
transformer_engine/pytorch/setup.py,sha256=qdLFWpteS0iGgNWzi8DAJjFJinZBWI6LMlyXzP9rgLs,2014
transformer_engine/pytorch/softmax.py,sha256=m-xaEG89MZKmejE2c4KVFiXO4NnrAhCOJ5-MetyHyQw,10244
transformer_engine/pytorch/transformer.py,sha256=VaGe3SCCeiECNkMzqx-GtWeBx2O9avgriq3tnyle790,39782
transformer_engine/pytorch/utils.py,sha256=kYfn4H_7bZ46eUREwA6bKtoaPj1Bpp6SfSMV-Dz5Pwc,12741
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=n8Sz4rFDy2tSM5nA1YKrKKPBU6r02ZXsbxtoSEoGWag,6867
transformer_engine/pytorch/dot_product_attention/__init__.py,sha256=YiIvGvL00zTFMl0IFeROia-c0IGD2JuZumRumu1OxFk,172
transformer_engine/pytorch/dot_product_attention/inference.py,sha256=F1fQKystl0LEQSgzeH1WSGnJML5uGseCb44qUlnuOrI,28772
transformer_engine/pytorch/dot_product_attention/rope.py,sha256=qO76o-S-06A9jCz7RK60EO2yjL4TK8AIiTKZvSq_fFY,8588
transformer_engine/pytorch/dot_product_attention/utils.py,sha256=01mWLrsWZ8IASIEcc51UsRIvUNAXNLjINIgyG-Ie5Gk,77665
transformer_engine/pytorch/module/__init__.py,sha256=JHc9_LIv9Uoe4N-vyiGxS5GspAu9WWl4fNWyVQP8mAE,506
transformer_engine/pytorch/module/_common.py,sha256=v1LcM95XcGcCQ7Q5flsmSP_kon59kAURRw_w8sxQZyI,7095
transformer_engine/pytorch/module/base.py,sha256=ZjMpotpfdgq5SCru-fu-HpM5_CjlCDj-677cW6dQmQs,46296
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=etX8phhBdbgFuzP0jEE-mllzsiqb5twgTUtz7yOeiic,26782
transformer_engine/pytorch/module/layernorm.py,sha256=E11uH1YpKsR78YneQ5LtWVbHD95E0Ps7S4KKvqQ8lT4,6542
transformer_engine/pytorch/module/layernorm_linear.py,sha256=vh9brR_2IXeH-rLApOwhP7J8kDNWpr2syTDXtuwSWhs,62975
transformer_engine/pytorch/module/layernorm_mlp.py,sha256=345zUGBN-hinVqUKt5WtQiZKifg1jk4_RD5yFtD1_KM,72881
transformer_engine/pytorch/module/linear.py,sha256=WRZsroOhnvrXBW2CXbFQiaQI2Kf1zASaJtgqIHBTJjs,54766
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=Y_PQK3iMCqFSCc4pkk8jYPMPOJPDY53NC3AFnh2dZr4,24761
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=zRE-jMa2NdUVXWarglsRXydwmn3pLpBk_WPved7YSBI,36888
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=Vbo1vHXuxwrFgHgdc9SI-nNjsk2L5p5cABmQCAw1HU4,32989
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=6hZjZFlkM82iMG7fMpvoRy3Y59By7Rb7mCr5CNtKhuA,1337
transformer_engine/pytorch/tensor/float8_tensor.py,sha256=Q8hN9AAVwmCc2hdAMblMMB6EPEMJNzf6zYn0rtiAhz8,27443
transformer_engine/pytorch/tensor/mxfp8_tensor.py,sha256=ZHnBMnN46sDriPGn8T812vduYENOKvpGOK3dsdj3i84,21006
transformer_engine/pytorch/tensor/quantized_tensor.py,sha256=w_lXGN5fhTLYQA_8qA-Alm6B0uGjsjk5ETXODDjIKUI,15549
transformer_engine/pytorch/tensor/utils.py,sha256=DWKp2pP5LReG1elQZr3Bb3wZE5VcDUsZ-VE1HL-1RP8,13354
transformer_engine/pytorch/tensor/_internal/__init__.py,sha256=WT49hCxtio4bYpUN75_wruxHTuDnyqOkZJnsnIKRgRI,176
transformer_engine/pytorch/tensor/_internal/float8_tensor_base.py,sha256=V-GX8mnNiIFaJTFecWQoTXmnvQd91osW4M_7SFc_UQ8,4745
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=3R7VomjWEZhRAhPWh18Gbttm7t0mgDxlVZUm2E36B9E,23870
transformer_engine-2.2.0.dist-info/licenses/LICENSE,sha256=vqG0Jm8UmfpLuxbW51d-a1tT0e8BCDWp1GCHYCLMhyI,10142
transformer_engine-2.2.0.dist-info/METADATA,sha256=Po7RJAveLmuBk9D4ugiuuS-KV1yrsHHeHKEZympudsA,16761
transformer_engine-2.2.0.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
transformer_engine-2.2.0.dist-info/top_level.txt,sha256=iWl1kDrBo4b5u2DyflPgpugikjyWbZ-_B3rrbSRoMnQ,19
transformer_engine-2.2.0.dist-info/RECORD,,
