Home Projects TransformerEngine
TransformerEngine

TransformerEngine

by NVIDIA · GitHub

A 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.

# cuda# deep-learning# fp4# fp8
View on GitHub
⭐ Stars
3.4k
🍴 Forks
774
🔥 Trending
+3 this week
📜 License
Apache-2.0
Commercial use OK
📅 Created
2022
🔄 Last commit
4 days ago
🏷️ Category
cuda
💻 Language
You maintain this project?

Claim its page: indexed whatever its rank, translated into six languages, and enriched with what you write yourself.

Claim this page →
TransformerEngine — GitHub preview card
📈 Star history
3.43k3.41k
2026-07-072026-07-22
📄 About

A 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.

Frequently asked questions

What is TransformerEngine?

A 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.

Is TransformerEngine open source?

TransformerEngine is an open-source project. It is released under the Apache-2.0 license.

Is TransformerEngine free?

Yes. TransformerEngine is free and open source — you can use, modify and self-host it.

🏅 Maintainer of this project?
OpenSourceAI badge — TransformerEngine

Add this live badge to your README — your GitHub stars and directory rank, refreshed daily.

[![OpenSourceAI](https://opensourceai.tech/badge.php?tool=nvidia-transformerengine)](https://opensourceai.tech/project/nvidia-transformerengine.html)
More badge options →
🧬 Related projects🧬 View the DNA map →