Skip to content

llm.engine

Training engine utilities.

This module provides wrappers for models, optimizers, and more to enable efficient training with mixed precision, distributed data parallelism, and more.