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.15
2.14.1
2.14
2.13
2.12
5 / 8 versions selected
Downloads (Last 6 months): 0

About

Summary

A library for accelerating Transformer models on NVIDIA GPUs.

Last Updated

May 8, 2026 at 17:58

License

Apache-2.0

Supported Platforms

linux-64

Home

https://github.com/NVIDIA/TransformerEngine

GitHub Repository

https://github.com/NVIDIA/TransformerEngine