country_code
Skip to main content
Ctrl+K
Transformer Engine 2.20.0.dev0 - Home Transformer Engine 2.20.0.dev0 - Home

Transformer Engine 2.20.0.dev0

  • GitHub
Transformer Engine 2.20.0.dev0 - Home Transformer Engine 2.20.0.dev0 - Home

Transformer Engine 2.20.0.dev0

  • GitHub

Table of Contents

  • Home

Getting Started

  • Installation
  • Getting Started
  • Frequently Asked Questions (FAQ)
  • Transformer Engine vX.YZ Release Notes

Python API documentation

  • Common API
  • Framework-specific API
    • PyTorch
    • Jax

Features

  • Low precision training
    • Introduction
    • Performance Considerations
    • FP8 Current Scaling
    • FP8 Delayed Scaling
    • FP8 Blockwise Scaling
    • MXFP8
    • NVFP4
    • GEMM Speedups Across Precisions
  • Other optimizations
    • CPU Offloading

Examples and Tutorials

  • Using FP8 and FP4 with Transformer Engine
  • Performance Optimizations
  • Accelerating Hugging Face Llama 2 and 3 Fine-Tuning with Transformer Engine
  • Accelerating Hugging Face Gemma Inference with Transformer Engine
  • Accelerating Hugging Face Mixtral MoE Fine-Tuning with Transformer Engine
  • Export to ONNX and inference using TensorRT
  • JAX: Integrating TransformerEngine into an existing framework
    • JAX: Dense GEMMs with TransformerEngine
    • JAX: Collective GEMMs with TransformerEngine
    • JAX: Attention with TransformerEngine
      • JAX: Single-GPU Attention with TransformerEngine
      • JAX: Context-Parallel Attention with TransformerEngine
    • JAX: Expert Parallelism with TransformerEngine
  • Operation fuser API
  • GEMM Profiling Tutorial

Advanced

  • C/C++ API
    • transformer_engine.h
    • activation.h
    • cast_transpose_noop.h
    • cast.h
    • cudnn.h
    • fused_attn.h
    • fused_rope.h
    • gemm.h
    • multi_tensor.h
    • normalization.h
    • padding.h
    • permutation.h
    • recipe.h
    • softmax.h
    • swizzle.h
    • transpose.h
  • Precision debug tools
    • Getting started
    • Config File Structure
    • API
      • Setup
      • Debug features
      • Calls to Nvidia-DL-Framework-Inspect
    • Distributed training
    • Adding custom feature to precision debug tools
  • Environment Variables
  • Attention Is All You Need!
  • Deep Dive into CP + THD + AG + Striped>1 + SWA support for Transformer Engine JAX
  • Low precision training
Is this page helpful?

Low precision training#

  • Introduction
    • Training in BF16/FP16
    • Lower precisions
  • Performance Considerations
    • Handling transposes
    • Memory usage
    • Fused layers
    • Distributed training
  • FP8 Current Scaling
    • FP8 data type
    • Scaling factors
    • Transpose handling
    • Distributed training
    • Supported devices
    • Examples
    • Developer Notes
      • All-gather of columnwise tensors
  • FP8 Delayed Scaling
    • Quantization with delayed scaling factors
    • Amax History Management
    • Distributed Training
    • Supported devices
  • FP8 Blockwise Scaling
    • Data Format
    • Handling transposes
    • Distributed training
    • Examples
    • Supported devices
    • Developer Notes
      • Swizzle of scaling factors
      • All-gather of columnwise tensors
  • MXFP8
    • Data Format
    • Handling transposes
    • Distributed training
    • Examples
    • Supported devices
    • Developer Notes
      • Swizzling scaling factors
      • All-gather of columnwise tensors
  • NVFP4
    • Data Format
    • Stochastic Rounding
    • Random Hadamard Transform
    • Handling transposes
    • Distributed training
    • Examples
    • Supported devices
    • Developer Notes
      • Swizzling scaling factors
      • All-gather of columnwise tensors
  • GEMM Speedups Across Precisions
    • Example: 5B Model on B300 (Blackwell)
    • Example: 5B Model on H200 (Hopper)
    • Speedup Is Shape-Dependent

previous

Jax

next

Introduction

NVIDIA NVIDIA
Privacy Policy | Your Privacy Choices | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved..