独角兽

训练Infra工程师

智谱AI

北京 / 上海 / 其他运维/架构/安全社会招聘

岗位描述

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