Skip to content

framework

NVIDIA Transformer Engine

NVIDIA's open-source library of kernels and low-precision recipes for training and serving Transformer models on its GPUs, with PyTorch and JAX front ends.

Known aliases

  • TE
  • Transformer Engine

Relationships

No evidence-backed relationships are recorded.

Current clusters