llm.engine llm/engine/__init__.py Training engine utilities. This module provides wrappers for models, optimizers, and more to enable efficient training with mixed precision, distributed data parallelism, and more.