composer

Composer 是一个基于 PyTorch 的开源深度学习训练框架，专注于提升模型训练的效率与可扩展性。它提供丰富的训练优化技术（如混合精度、梯度裁剪、重计算、分层学习率、数据增强策略等），并原生支持多节点分布式训练、自动并行配置和内存/计算效率优化，显著降低大规模模型（如 LLM、扩散模型、Transformer）的训练成本与开发复杂度。适用于需要快速迭代、高效利用 GPU 集群进行中大型神经网络训练的研究与工程场景，尤其适合对训练速度、资源利用率和实验可复现性有较高要求的团队。