灵波科技-具身智能 Infra 优化工程师(实习)当面告知
蚂蚁科技集团股份有限公司
2026-12-12 00:00:00
27届28届实习本科
岗位详情
地区:
上海
专业要求:
自动化计算机科学与技术软件工程
岗位职责:
1. 参与具身智能大模型基础设施建设,支持 VLA/VLM、机器人策略模型的训练、推理与部署;
2. 协同算法团队适配 CT、SFT、RL 等训练任务,完善分布式训练、数据 IO、Checkpoint 及任务调度能力;
3. 参与训练性能与稳定性优化,包括计算图、计算通信重叠、显存和资源利用率优化;
4. 参与模型推理链路建设,包括模型导出、量化、TensorRT/TRT-LLM 编译、服务化部署及性能分析;
5. 针对延迟、吞吐、显存占用和训练效率等指标开展端到端性能优化。
招聘要求:
1. 计算机、自动化、电子、软件工程等相关专业,本科及以上在读;
2. 具备良好的编程能力,熟悉 Python;掌握 C/C++ 或 CUDA 者优先;
3. 熟悉 Linux,具备良好的数据结构、算法、操作系统和计算机体系结构基础;
4. 对 LLM、VLM、VLA、Diffusion 或机器人学习有基本了解,并有相关训练或部署实践;
5. 了解 PyTorch 及 FSDP、Megatron、VeRL 等至少一种分布式训练框架,或 TensorRT、TRT-LLM、Triton 等推理优化工具;
6. 具备较强的问题定位、工程实现和团队协作能力。
加分项:
1. 有 CUDA Kernel、TensorRT Plugin、Triton Kernel 或 NCCL 通信优化经验;
2. 有模型量化、Attention、KV Cache、视觉编码器或动作头优化经验;
3. 有大规模训练稳定性、GPU 性能分析、异构资源调度或内存优化经验;
4. 有具身智能、机器人或多模态模型训练与部署经验;
5. 有 ICPC、ASC、OI、MO 等竞赛经历或相关开源项目贡献。
公司介绍

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