CMD + K
conda-forge/transformer-engine-torch

transformer-engine-torch

Community

A library for accelerating Transformer models on NVIDIA GPUs.

Copied fromcf-post-staging / transformer-engine-torch

Installation

To install this package, run one of the following:

Conda
$conda install conda-forge::transformer-engine-torch

Usage Tracking

2.5
2.3
2.2
2.1
2.0
5 / 8 versions selected
Total downloads: 0

About

Summary

A library for accelerating Transformer models on NVIDIA GPUs.

Information Last Updated

Dec 5, 2025 at 08:46

License

Apache-2.0

Total Downloads

17.0K

Platforms

Linux 64 Version: 2.5

Home

https://github.com/NVIDIA/TransformerEngine

GitHub Repository

https://github.com/NVIDIA/TransformerEngine