NVIDIA/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.
GitHub repository with 3,379 stars and 739 forks.
Language: Python
Topics: cuda, deep-learning, gpu, machine-learning, python, pytorch, fp8, jax, fp4