招聘中 实习

平台技术事业群 - 大模型训推Infra实习生

蚂蚁集团

  • 杭州、广州
  • 技术类

收录时间

岗位职责

  1. 训推框架全栈优化:负责大模型分布式训练与推理框架研发,推进 MoE 架构与框架协同设计(Co-design),实现千亿级模型的高效训推;深入国产卡适配与算子优化,持续提升吞吐、效率及资源利用率。
  2. 核心系统性能攻关:研发 KVPool 内存池与高性能网络传输机制,优化 NUMA 亲和性、RDMA 零拷贝及计算通信重叠(Overlap),解决跨节点 KV Cache 传输与内存管理瓶颈。
  3. 智能剖析工具建设:主导 XPU-Profiler 模块开发与产品化,完善采集分析能力;探索 AI 辅助研发(RSI 实验)新范式,沉淀可复用的性能分析方法与工具。
  4. 系统集成与稳定性:推动与 vLLM、SGLang 等主流引擎的深度集成,建设全链路可观测性与性能测试体系,保障高负载场景下的系统稳定运行。

职位要求

  1. 计算机、软件工程或相关专业的非应届在读生,每周可实习 ≥3 天,连续 ≥3 个月。
  2. 具备扎实的数据结构、操作系统、计算机网络基础;熟悉 C++ 或 Python,具备 Linux 环境下的开发、调试能力,有良好的编码习惯。
  3. 了解模型训练、推理基本流程及 PyTorch 等框架,对 CPU、GPU 执行机制或分布式系统有兴趣;或对 KV Cache、Attention、Prefill/Decode 等概念有一定认识,对 AI 系统性能优化有兴趣。
  4. 具备良好的问题分析和动手能力、团队协作能力,具有强烈的好奇心和自驱力; 加分项:
  5. 以下方向有实践经验者(任一):RDMA / 高性能网络通信、CUDA / GPU 性能优化、国产卡推理和后训练、内存管理 / 分布式缓存、vLLM / SGLang / Mooncake / nixl / LMCache 等开源项目。
  6. 有 CUDA/Triton、GPU Kernel 优化、Profiler/CUPTI/eBPF、分布式训练与推理、NCCL 通信分析,或 AI Agent、自动化实验与评测工具开发经验者优先,欢迎提供课程、科研或开源项目的代码和实验材料。
  7. 有系统性能优化或开源贡献经历者优先。

相似岗位

有些机会,只在官网短暂出现

真正值得关注的岗位,常常只在企业官网短暂开放,可能两三天后就下线,也未必会同步到综合招聘平台。没有持续关注,你甚至不会知道它曾经出现。职先机持续聚合并核验官网岗位,帮你抓住职场先机,快人一步。

微信小程序 / 当前岗位

在微信里继续看这个岗位

蚂蚁集团平台技术事业群 - 大模型训推Infra实习生

扫码直达当前岗位收藏、浏览记录和下线提醒留在微信里
电脑端可直接微信扫码;手机端可保存小程序码后在微信中识别。
微信扫码在职先机小程序查看平台技术事业群 - 大模型训推Infra实习生小程序码按需加载
职先机微信小程序一岗一码 · 正式版直达

岗位提醒 · 01

收藏当前岗位,下线及时通知

当前岗位平台技术事业群 - 大模型训推Infra实习生蚂蚁集团 · 杭州、广州

01

扫码进入当前岗位无需重新搜索,直接打开当前岗位详情。

02

收藏并开启提醒在小程序内收藏后,岗位下线或变化时及时通知。

03

继续查看相似机会原岗位变化时,可继续浏览相关在招岗位。

职先机会持续核验公开岗位;岗位状态与最终招聘结果仍以企业招聘官网为准。

微信小程序扫码打开当前岗位

扫码打开当前岗位,完成收藏并开启下线提醒。

微信扫码在职先机小程序收藏平台技术事业群 - 大模型训推Infra实习生小程序码按需加载

微信小程序 · 当前岗位

请前往微信小程序提交内推申请

蚂蚁集团平台技术事业群 - 大模型训推Infra实习生实习 · 杭州、广州

01进入当前岗位扫码或打开后直达本岗位内推申请页。

02提交 PDF 简历简历替换、处理进度和消息提醒在小程序内同步。

手机端将尝试直接打开微信;电脑端请使用微信扫码。

微信扫码进入平台技术事业群 - 大模型训推Infra实习生内推申请页小程序码按需加载
当前岗位专属码扫码后无需重新搜索岗位

内推提示

暂不支持该岗位内推

如需投递,建议前往官网进行投递。