蚂蚁集团-训推系统研发专家-杭州/北京/上海当面告知
蚂蚁科技集团股份有限公司
2027-01-03 00:00:00
社招本科
岗位详情
地区：
北京
·
上海
·
杭州市
专业要求：
专业不限
岗位职责：
蚂蚁ASystem致力于打造下一代AI基础软件，并基于下一代的AI基础软件寻找通用智能的新方法，追求智能上限。
1、负责训推一体框架的设计与开发，服务蚂蚁内部的强化学习场景；
2、建设面向训推一体的显存管理体系和高性能数据存储方案；
3、负责实时高性能训推系统设计与开发，如分布式训练加速策略、算子融合、编译优化、模型量化、混合精度、异构硬件加速等；
4、负责整体性能优化与架构升级，持续提升训练/推理性能；
5、与算法工程师深度合作，为重点项目进行算法与系统的联合优化。
招聘要求：
1. 精通至少一门 Python/Go/C++ 等编程语言，并有良好的代码风格；
2. 具有分布式系统服务/并行计算系统设计与研发优化经验；
3. 代码级精通主流深度学习框架及扩展库的使用及算子开发，例如TensorFlow/PyTorch/Megatron/Deepspeed/vLLM/Sglang等；
4. 好奇心强，热爱技术且对强化学习领域有深入钻研优先；
加分项：
1、熟悉至少一种主流的RLHF框架，如OpenRLHF/veRL/AReal/ChatLearn等；
2、熟悉Ray框架或其他强化学习相关计算框架；
3、在计算机系统网络顶会OSDI/SOSP/NSDI/ATC/EuroSys上有文章发表经验
公司介绍
蚂蚁科技集团股份有限公司
蚂蚁科技集团股份有限公司，成立于2000-10-19，位于浙江，法定代表人：韩歆毅，登记状态：存续，注册资本3500000万人民币，公司类型：民营/私企
蚂蚁集团是中国较大的移动支付平台支付宝的母公司，也是全球金融科技开放平台，致力于以科技推动包括金融服务业在内的全球现代服务业的数字化升级，携手合作伙伴为消费者和小微企业提供普惠、绿色、可持续的服务，为世界带来微小而美好的改变。