AI infra实习生-大安全当面告知
蚂蚁科技集团股份有限公司
2026-12-12 00:00:00
27届28届实习本科
岗位详情
地区:
上海
专业要求:
人工智能计算机科学与技术软件工程
岗位职责:
部门介绍:
我们正在建设面向大模型时代的 AI Infra,支撑万卡级算力集群和真实业务模型训练,覆盖 CPT、SFT、RL、MoE、多模态等场景。这里不是实验室里的小规模 Demo,而是真实的大模型训练负载和生产级复杂环境。你会接触分布式训练、通信优化、算子优化、容错恢复、精度保障、资源调度、异构芯片适配等核心问题,也有机会参与新模型、新架构与训练系统的 Algorithm-System Co-design。
我们希望吸引对 AI Infra 有兴趣、基础扎实、动手能力强的同学,在真实项目中一起解决有挑战的问题。
职位描述:
1.参与大模型训练框架和训练基础设施研发,支持 CPT、SFT、RL、MoE、长上下文及多模态训练;
2.参与 PyTorch、Megatron、FSDP、verl 等训练体系的适配和优化,学习并实践 TP、PP、DP、EP 等分布式并行方案;
3.参与大规模训练中的性能分析和优化,包括计算、通信、显存、长尾及资源利用率等问题;
4.参与 Checkpoint、断点续训、故障恢复、精度比对、可观测等稳定性能力建设;
5.参与国产及异构 AI 芯片的训练适配与性能优化。
招聘要求:
职位要求:
1. 熟悉 Python、C++ 中至少一种语言,具备良好的系统设计和工程实现能力;
2. 熟悉 PyTorch 训练机制,对分布式训练、并行策略、显存优化或高性能计算有实践经验;
3. 对 Megatron-LM、FSDP、verl 等框架中的一种或多种有深入理解;
4. 熟悉 NCCL、RDMA、集合通信、GPU/NPU 算子优化者优先;
5. 有大模型预训练、MoE、长序列、强化学习训练或千卡级集群实践经验者优先;
6. 喜欢解决真正困难的问题,愿意从模型、框架、系统一直深入到芯片和网络。
公司介绍

蚂蚁科技集团股份有限公司
蚂蚁科技集团股份有限公司,成立于2000-10-19,位于浙江,法定代表人:韩歆毅,登记状态:存续,注册资本3500000万人民币,公司类型:民营/私企
蚂蚁集团是中国较大的移动支付平台支付宝的母公司,也是全球金融科技开放平台,致力于以科技推动包括金融服务业在内的全球现代服务业的数字化升级,携手合作伙伴为消费者和小微企业提供普惠、绿色、可持续的服务,为世界带来微小而美好的改变。