Wheelodex » Projects » transformer-engine

transformer-engine

View on PyPI — Reverse Dependencies (2)

2.1.0 transformer_engine-2.1.0-py3-none-any.whl

Wheel Details

Project: transformer-engine
Version: 2.1.0
Filename: transformer_engine-2.1.0-py3-none-any.whl
Download: [link]
Size: 402051
MD5: de9be8b51e4b374f217f83c04e9ba7fb
SHA256: c3253024f7426d67d9defb14bd395cee6b07ed3d28a5fb959575118dc3b4366e
Uploaded: 2025-03-13 10:29:50 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.2
Name: transformer_engine
Version: 2.1.0
Summary: Transformer acceleration library
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.8, <3.13
Requires-Dist: transformer_engine_cu12 (==2.1.0)
Requires-Dist: transformer_engine_torch (==2.1.0); extra == "pytorch"
Requires-Dist: transformer_engine_jax (==2.1.0); extra == "jax"
Provides-Extra: pytorch
Provides-Extra: jax
Description-Content-Type: text/x-rst
Dynamic: classifier
Dynamic: description
Dynamic: description-content-type
Dynamic: provides-extra
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary
License-File: LICENSE
[Description omitted; length: 15908 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.8.2)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
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=gVTZ-QaN1luwh5W02ZclPd0w2_q-nuGugs6JFaS_jFw 8041
transformer_engine/jax/__init__.py sha256=naPxN0LMpBD0jambTknUsmqqSPfi2nMiFLdJ6m4jf-E 3365
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=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=q-k8M3P3AhGHkSJ8On0BR28QjiP7mbnfaJYYgx3Xkus 17552
transformer_engine/jax/cpp_extensions/attention.py sha256=xgO5I1AeoL-Wrqg6gRy2sj81wdo3KSJFNNGei64WANs 94279
transformer_engine/jax/cpp_extensions/base.py sha256=4fR61FZC7vhEHYDfxBmcLKZrqqefEAoSRrslT9C-6zM 3460
transformer_engine/jax/cpp_extensions/custom_call.py sha256=t1qTXPmeMylZxlz9qfBzVmOyapU64wCN5EdOmXv4zsk 3363
transformer_engine/jax/cpp_extensions/misc.py sha256=tPehlTbxK4EXunQPMO6T9uQlWEZ1lbPB0CNmM1opvJc 6033
transformer_engine/jax/cpp_extensions/normalization.py sha256=YHXGYP8UTyIP5Qsj7dzzQfs6vXtwTmyc0n7eaFdYcSE 56557
transformer_engine/jax/cpp_extensions/quantization.py sha256=FKr2sfAKV9pkisrQFd03hEWWhGF4FI_Omw15KUhatgc 6583
transformer_engine/jax/cpp_extensions/softmax.py sha256=vaSP1gMMNbojuGVooeh5Tn4yCQC7gC9QvKHcfpGnX2E 32958
transformer_engine/jax/cpp_extensions/transpose.py sha256=cJJN6VLf_MGPVWWiGKkSZYMTBkmG5TS2fuBd4HWwMeQ 45925
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=kdusYJzzvKgisLr3k3V8sRylNJFdfqd1M5OGTudlMbQ 4077
transformer_engine/pytorch/attention.py sha256=7EgyTVMUX9Bn4w2z8yWP5xAfU2QztSISEJLy9uCQkBA 391288
transformer_engine/pytorch/constants.py sha256=frX1yT9smiWgu4trKuRZ9UOpY0UKLR1Umk9CEDXYAXg 1368
transformer_engine/pytorch/cpu_offload.py sha256=qt3TrxJqrmAvUwOhqchQjPOlqWqx03_f6drA82Vz3E8 20780
transformer_engine/pytorch/distributed.py sha256=G_GEPP0nDF4OeFjK9zXIWeglvPIr5-YE1jhjg1zjkJY 44361
transformer_engine/pytorch/float8_tensor.py sha256=KdHMbjsSpPWkB1Cu4d0Z01QhATMijK03e_XCl1p88fo 232
transformer_engine/pytorch/fp8.py sha256=42fCWJj0DGxWGEM6C6VQitU4UWN0UjSCqvUQJnjaBHQ 30244
transformer_engine/pytorch/graph.py sha256=vPCLSITHkDiGEylMER8LMOMsVBJ7ZyQlBgta6wHvxrA 32984
transformer_engine/pytorch/jit.py sha256=Ucf6l2Ifp77yEnSJzqIHYaUBmyOrlYyvtA_HyWQYqwo 9084
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=-VwzTZklrSqbzSWD4YkgSw_y6m9Sa6TIgOE1-oncFmw 39324
transformer_engine/pytorch/utils.py sha256=QBm3K2DsEg1E7NFLL6ptdWDwPoFFwtG3BhCtlELfqrU 12624
transformer_engine/pytorch/cpp_extensions/__init__.py sha256=qZ1qHrZN_Du1jhkefQx1PtQFRAGsvPrpOR-b5uUhoIE 251
transformer_engine/pytorch/cpp_extensions/fused_attn.py sha256=gy6N1gOkysWwMCx-bvKdxUyDZTU-K2tBs6xNgL9gf4I 18660
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=jb2c9VlyefCYUnDp5TkJs88XSIOf-vHM4c05DFYijv8 7577
transformer_engine/pytorch/module/base.py sha256=4PCMizncCGfudel5m4j8Za7nMWuWUptOSuIhiw4FNmc 46344
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=IxNKTxiSdu44mmsHTVeF_2bJHmBP-DHHzcCmiJKQFyw 55984
transformer_engine/pytorch/module/layernorm_mlp.py sha256=SapXtEF4Hg8k7w6NHBg7A8vCaWeFczykZdCaxqTCw3I 64643
transformer_engine/pytorch/module/linear.py sha256=0xz9a6tU12OMAlk16B1JqOLiGp57hRgVhFV7FWo7iNY 46820
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=YM-ygXKwE5LvC0vM5cJzf52ASimdm5dy6eCyPFZ0Au8 4207
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=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=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=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=kcDZLFryVXYFKhNCMcRjw_Jqk2sXmjSdPRVtCK6fi60 23201
transformer_engine-2.1.0.dist-info/LICENSE sha256=vqG0Jm8UmfpLuxbW51d-a1tT0e8BCDWp1GCHYCLMhyI 10142
transformer_engine-2.1.0.dist-info/METADATA sha256=izvm9PAHLvdCjPIzVCZkQ3Nr24LVzCRlpU8M4Y1giYQ 16737
transformer_engine-2.1.0.dist-info/WHEEL sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM 91
transformer_engine-2.1.0.dist-info/top_level.txt sha256=iWl1kDrBo4b5u2DyflPgpugikjyWbZ-_B3rrbSRoMnQ 19
transformer_engine-2.1.0.dist-info/RECORD — —

top_level.txt

transformer_engine