蚂蚁集团-异构计算研发工程师-北京/杭州当面告知
蚂蚁科技集团股份有限公司
2026-11-29 00:00:00
其他社招
岗位详情
地区：
北京
·
杭州市
专业要求：
计算机类
岗位职责：
蚂蚁DeepXPU团队专注在基于GPU等AI芯片的高性能异构计算系统研发，欢迎加入我们一起致力于：
● Software&Hardware Co-Design：深入分析GPU等AI芯片的硬件架构，打磨极致高效的算子与显存优化技术，优化大模型引擎。通过系统建设和优化，推动计算效率逼近硬件的性能极限，打造面向大模型的AI系统生态，提升智算集群的整体效率。
● Algorithm&System Co-Design：从系统的视角出发，和算法团队紧密配合，共同探索大模型结构和范式的创新，并面向大规模AI应用场景进行系统级优化与落地。
招聘要求：
● 计算机相关学科专业，基本功扎实，3年以上相关工作经验。
● 熟悉Linux开发环境，掌握C/C++/Python编程语言，具备CUDA算子开发能力，掌握cutlass/CuTe/triton等算子开发工具，掌握nsys/ncu等profile工具。
● 深度理解GPU体系结构和CUDA编程模型，掌握GPU算子开发和性能调优技术，掌握GPU显存优化技术。
● 掌握FlashAttention/FlashMLA/DeepGEMM/FlashInfer等项目源码，有该加速库优化经验者优先。
● 掌握Megatron-LM或vllm/sglang等训推框架，有实际性能调优经验和开源代码贡献者优先。
● 有AI编译器开发和优化经验者优先。
● 有国产加速卡程序开发和优化经验者优先。
● 有开源贡献、系统顶会论文或者相关专利者优先。
● 有创新想法和技术攻关能力，优秀的逻辑分析和解决问题的能力，团队沟通协作能力强。
公司介绍
蚂蚁科技集团股份有限公司
蚂蚁科技集团股份有限公司，成立于2000-10-19，位于浙江，法定代表人：韩歆毅，登记状态：存续，注册资本3500000万人民币，公司类型：民营/私企
蚂蚁集团是中国较大的移动支付平台支付宝的母公司，也是全球金融科技开放平台，致力于以科技推动包括金融服务业在内的全球现代服务业的数字化升级，携手合作伙伴为消费者和小微企业提供普惠、绿色、可持续的服务，为世界带来微小而美好的改变。