蚂蚁集团 · 社会招聘
蚂蚁数字科技-数科技术部-大模型强化学习研发专家
杭州 ·2年以上·本科
上海·杭州
薪资面议
前往 蚂蚁集团 官网投递
职位描述
1.负责实时高性能强化学习框架的设计与开发,服务蚂蚁数科的模型训练场景,如分布式训练加速策略、显存管理、算子融合、编译优化、模型量化、混合精度、异构硬件加速等;
2.与算法工程师深度合作,针对大模型强化学习训练和推理场景,进行端到端性能分析和优化,包括分布式训练加速、显存优化、通信效率提升等,满足大规模 Agent 训练的性能要求。
【任职要求】
1.具备大模型训练框架或分布式推理框架开发经验,熟悉 Megatron-LM、DeepSpeed 等分布式训练框架,或具备 SGLang、vLLM 等大模型分布式推理与推理加速框架开发经验者优先;
2.具备大模型强化学习框架相关经验者优先,例如 AReaL、veRL、Slime 等, 对RL流程熟悉者优先;
3.具备良好的逻辑分析能力,对复杂框架的模块设计、抽象边界和工程实现有一定理解,热爱技术且对强化学习领域有深入钻研优先;
4.熟悉 Agent 工程或 AI Coding Agent 工具链者优先;
5.有大模型训练经验优先。
最后确认在招:2026-09-01 19:09:33 · 来源平台:alibaba