顶尖实习-大模型训练框架研发工程师-AI基建

职位描述

1.AI训练框架的设计与演进 2.通信、显存与数据通路的极端性能优化 3.算法与工程协同优化

任职要求

1.参与大模型（包括但不限于语言模型、多模态模型）的预训练、有监督微调、强化学习对齐，分布式推理等全流程算法实现与实验。 2.协助进行大规模数据处理、分布式训练策略优化、训练和推理性能诊断与调优（如吞吐、显存、通信效率）。 3.参与研究并实现模型训练与推理的性能优化方案，探索前沿的高效训练和分布式推理技术。 4.跟踪并复现学术界与工业界在模型架构、训练和推理算法及系统优化上的最新进展。 5.配合团队完成训练与推理任务部署、问题排查及技术文档编写。