logo of antgroup

蚂蚁金服【蚂蚁星】AI Infra 训练引擎 / 训练框架研发工程师-27届

校招全职2027届蚂蚁星- Plan A人才计划地点:北京 | 上海 | 杭州状态:招聘

工作描述


任职要求
1.熟悉 PythonC++ 中至少一种语言,具备良好的系统设计和工程实现能力;
2. 熟悉 PyTorch 训练机制,对分布式训练、并行策略、显存优化或高性能计算有实践经验;
3. 对 Megatron-LM、FSDP、verl 等框架中的一种或多种有深入理解;
4. 熟悉 NCCL/HCCL、RDMA、集合通信、GPU/NPU 算子优化者优先;
5. 有大模型预训练、MoE、长序列、强化学习训练或千卡级集群实践经验者优先;
6. 喜欢解决真正困难的问题,愿意从模型、框架、系统一直深入到芯片和网络。

工作职责
部门介绍:
在万卡规模上,重新定义大模型训练基础设施。我们正在建设面向大模型时代的 AI Infra,支撑超大规模算力集群稳定运行,覆盖…
登录查看完整工作描述
微信扫码,1秒登录

包括英文材料
Python+
C+++
系统设计+
PyTorch+
还有更多 •••
相关职位

logo of antgroup
实习蚂蚁星- Pla

1. 计算机科学、人工智能等相关专业硕士及以上学历,熟悉 Transformer 架构、Agen Mem系统及GRPO/PPO/DPO/SFT等算法; 2. 在分布式系统领域或者并行计算领域有研究和实

更新于 2026-05-12杭州
logo of antgroup
校招2027届蚂蚁星

1. 计算机科学、人工智能等相关专业硕士及以上学历,熟悉 Transformer 架构、Agen Mem系统及GRPO/PPO/DPO/SFT等算法; 2. 在分布式系统领域或者并行计算领域有研究和实

更新于 2026-05-14杭州
logo of antgroup
校招2027届蚂蚁星

1. 核心参与过业界知名的开源评测项目或 Benchmark 建设; 2. 在 NeurIPS、ICML、ACL、EMNLP 等相关顶会发表过高质量论文; 3. 优秀的数据敏感度与逻辑分析能力,能从复

更新于 2026-05-14杭州
logo of antgroup
实习蚂蚁星- Pla

1. 核心参与过业界知名的开源评测项目或 Benchmark 建设; 2. 在 NeurIPS、ICML、ACL、EMNLP 等相关顶会发表过高质量论文; 3. 优秀的数据敏感度与逻辑分析能力,能从复

更新于 2026-03-26杭州