TransformerEngine

A library for accelerating transformer-based model training and inference on Nvidia GPUs.

1company uses it

Companies that use TransformerEngine