蚂蚁集团-具身智能推理优化与算子开发工程师-上海当面告知
蚂蚁科技集团股份有限公司
2026-11-22 00:00:00
社招本科硕士
岗位详情
地区：
上海
专业要求：
计算机科学与技术
岗位职责：
1. 面向具身智能、多模态 VLA/VLM、机器人策略模型等推理场景，设计并实现高性能 CUDA/C++ 算子。
2. 针对 H系列 GPU 与 RTX 4090 等硬件平台，进行端到端推理性能优化，包括延迟、吞吐、显存占用和稳定性优化。
3. 打通具身模型推理机部署流程，支持模型导出、量化、TensorRT/TRT-LLM 编译、服务化部署与性能 profiling。
4. 有 LLM/VLM/VLA 推理链路中的关键模块的优化经验，包括 attention、KV cache、视觉编码器、动作头、后处理等。
5. 针对跨卡、跨机推理通信场景，开发和优化 NCCL/RDMA/NVLink/PCIe 相关通信算子与调度策略。
招聘要求：
1. 计算机及相关专业，本科及以上学历。
2. 具备扎实的 C/C++ 编程能力，良好的数据结构和算法基础。
3. 理解操作系统、计算机体系结构和 GPU 并行计算基础。
4. 熟悉 CUDA 编程，有性能分析和优化经验。
5. 熟悉 TensorRT/TRT-LLM，或其他 LLM 推理优化相关算子库。
6. 具备良好的团队沟通、协作能力和责任心。
-加分项：
1. 有 ICPC、ASC、OI、MO 等竞赛获奖经历。
2. 有 CUDA kernel、TensorRT plugin、Triton kernel 或通信算子开发经验。
3. 有具身智能、机器人、多模态模型推理部署经验。
公司介绍
蚂蚁科技集团股份有限公司
蚂蚁科技集团股份有限公司，成立于2000-10-19，位于浙江，法定代表人：韩歆毅，登记状态：存续，注册资本3500000万人民币，公司类型：民营/私企
蚂蚁集团是中国较大的移动支付平台支付宝的母公司，也是全球金融科技开放平台，致力于以科技推动包括金融服务业在内的全球现代服务业的数字化升级，携手合作伙伴为消费者和小微企业提供普惠、绿色、可持续的服务，为世界带来微小而美好的改变。