蚂蚁集团
蚂蚁集团-算子开发工程师-上海/北京
蚂蚁集团-算子开发工程师-上海/北京
发布于 3 小时前普通员工/个人贡献者
全职员工仅现场办公算法优化工程师
算法优化工程师
Cuda
Gpu编程
Rocm
性能优化
算子开发
计算机架构
值不值得投
71综合评分
综合评分 71
前沿芯片算子开发，技术深度极高、成长性强，但WLB和办公模式信息未知。
从薪资福利、成长空间、工作节奏和岗位方向综合评估，方便横向比较。
薪资怎么样
4万
与算法优化工程师中位数基本持平
发布情况
新岗位 · 今天发布
公司情况
这家公司招聘很活跃
约 840 个在招 · 其中算法优化工程师 6 个
市场机会
选择面较广
上海 · 算法优化工程师 · 约 150 个在招
职位详情
关于这个职位
该职位负责基于蚂蚁自研计算芯片，进行高性能AI算子的研发与性能调优
你需要编写高效的计算核与数据流，交付性能极致的算子，并持续迭代算子库及其工具链，与业务团队共同构建技术生态
最低要求
硕士及以上学历，计算机/电子/通信等专业，4年以上相关工作经验，有AI算子开发经验者优先
熟悉Linux开发环境，掌握Python/C++等语言，具备良好的编程基础以及系统设计能力
熟悉GPU或SIMD编程体系结构和原理，对Cuda、ROCm算子具备一定的开发能力
熟悉计算机架构，善于分析关键路径
工作职责
基于自研计算芯片，研发高性能算子、对算子和应用进行调优
针对业务落地过程中的性能与泛化性问题，不断迭代算子库
能够编写高效的计算核和数据流，交付性能极致的算子
不断迭代算子工具链，以确保其完整性和易用性，和业务一起构建生态
优先资格
有线性代数和概率论基础，熟悉数学建模、算法设计和优化方法者优先
有密码算法经验者优先
AI 洞察
优缺点分析
优点
- 参与自研芯片生态构建，技术栈前沿且独特，能积累稀缺的底层优化经验
- 工作直接影响AI应用的性能表现，技术价值高，成果易于衡量
- 身处头部科技公司，有丰富的技术资源和协作平台
- 技术门槛极高，需要同时精通芯片架构、算法和高性能编程，学习曲线陡峭
- 需要与算法、业务等多团队紧密沟通协作，对综合能力要求高
- 适合对计算机底层原理有浓厚兴趣、乐于追求极致性能、具备扎实编程功底和系统思维，并希望深耕AI基础设施领域的工程师
缺点 / 挑战
- 性能优化工作往往面临极限挑战，需要解决复杂且具体的技术难题
角色解读
- 技术专家路线：深耕芯片算子领域，成为性能优化和架构设计的权威
- 架构师路线：从单个算子开发扩展到整个计算库或芯片软件栈的架构设计
- 技术管理路线：带领算子开发团队，规划技术方向，协调跨团队合作
- 在自研AI芯片上设计和开发高性能计算算子，是连接硬件能力与上层AI应用的关键环节
- 对算子及使用算子的应用进行性能分析和极致调优，确保计算效率最大化
- 负责算子工具链的开发与维护，提升开发体验和生态易用性，支持业务快速落地
- 深厚的GPU架构和编程知识，熟练使用Cuda或ROCm进行算子开发
- 扎实的C++和Python编程能力，以及系统级设计和分析能力
- 对计算机体系结构有深刻理解，能够进行关键性能路径分析
申请策略
- 面试前务必了解蚂蚁集团在AI芯片或计算领域的公开信息和技术动向
- 准备1-2个体现你解决复杂性能问题或进行系统优化的深度技术案例
- 重点突出使用Cuda/ROCm进行算子开发、优化的具体项目经验，并量化性能提升结果
- 详细描述在复杂系统或大规模计算场景下的性能分析、调试和优化经历
- 展示对GPU体系结构、并行计算、内存层级等底层概念的理解深度
- 深入学习Cuda/ROCm编程模型和高级优化技巧，可借助官方文档和开源算子项目实践
- 复习和加深对计算机体系结构、编译原理、高性能计算相关知识的理解
- 尝试阅读和分析知名深度学习框架（如PyTorch/TensorFlow）中的算子实现
面试指南
- 对于性能优化问题，采用“定位瓶颈-分析原理-提出方案-验证效果”的结构回答，并强调数据和工具的使用
- 对于设计类问题，从功能、性能、可维护性、可扩展性等多个维度阐述思考，并联系具体技术细节
- 结合自身经验，说明如何在实际项目中平衡性能、开发效率和业务需求
- 请描述你优化过的一个最复杂的CUDA算子，遇到了哪些性能瓶颈？你是如何分析并解决的？
- 在设计一个新的计算算子时，你如何考虑其在不同硬件配置（如不同GPU型号）上的性能和泛化性？
- 你认为构建一个易用、完整的算子工具链需要包含哪些关键部分？你会如何设计？
- 如果一个算子在GPU上运行速度未达预期，你会使用哪些工具和方法来定位问题？
- 复习Cuda编程的核心概念（线程层次、共享内存、合并访问等）和性能分析工具（如Nsight Systems/Compute）
职位点评
薪资福利75
成长发展90
工作生活40
使命价值70
薪资福利
75中等
该职位属于核心技术岗，对技能和经验要求高，预计薪酬具有市场竞争力。但JD未提及具体福利，对稳定性和福利的满足程度取决于公司政策。
薪资信号市场水准 (3千–5万/月)
成长发展
90较高
职位处于AI计算前沿，涉及自研芯片底层开发，技术深度和独特性极强，能为从业者提供非常稀缺且高价值的技术积累和成长机会。
技术前沿前沿/新兴技术
技术栈Cuda、ROCm、GPU编程、算子开发、计算机架构、自研芯片
业务类型profit_center
工作生活
40较低
JD未明确工作模式和办公地点类型，且涉及芯片底层开发，通常需要现场协作和调试，工作强度和压力可能较大。
工作模式未明确
办公地点科技园/产业园
加班情况未提及（无法判断）
使命价值
70中等
参与自研芯片和算子生态构建，对推动国产AI算力发展具有技术意义。但JD未突出明确的社会使命或愿景，更多聚焦于具体技术实现。
行业发展高速增长赛道
社会影响中性/一般
创新程度积极采用新技术
Watch Jobs