AI院-训练Infra工程师

1、负责大规模预训练框架的研发、优化和维护，根据业务需求持续改进训练框架和策略，提升模型训练效率 2、分析和定位训练中的性能瓶颈，实施针对性优化措施，提升训练效率和稳定性 3、跟进业界技术进展，不断同步与集成最新训练优化策略 【任职要求】 1、对自然语言处理、计算机视觉和多模态算法有深入理解，熟悉主流的 LLM 和 VLM 模型架构，有分布式训练经验 2、精通 Python 编程语言，熟悉 PyTorch 深度学习框架和 Megatron 分布式训练框架 3、有大规模预训练优化 / MoE 训练优化经验的优先考虑 加分项： 1、在ACM、NOI、IOI、超算比赛中有获奖经历者优先 2、具有 cuda 算子优化/profiling 能力者优先