Transformer Engine
Release: v2.21.0 (2026-11-10)Commit: 7457e47License: Apache-2.0A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.
$ pip install --no-build-isolation "transformer_engine[pytorch]"