返回岗位列表

AI加速器架构研发工程师

查看原岗位

地点：北京薪资：薪资未公开日期：2024-10-22

岗位摘要

基于现有指令集和编译框架，设计并实现优化深度学习算子，完成对PyTorch等主流框架的适配；负责AI加速器软硬件协同设计与优化，包括ISA设计、微架构设计及性能分析；通过仿真器进行架构探索，分析性能瓶颈，结合模型算法探索优化方向

岗位职责

基于现有指令集和编译框架，设计并实现优化深度学习算子，完成对PyTorch等主流框架的适配
负责AI加速器软硬件协同设计与优化，包括ISA设计、微架构设计及性能分析
通过仿真器进行架构探索，分析性能瓶颈，结合模型算法探索优化方向
与合作伙伴紧密配合，参与硬件架构设计与迭代，实现算法到硬件的高效映射
追踪和研究AI芯片领域的前沿技术发展

任职要求

计算机科学、电子工程、集成电路等相关专业博士学位
扎实的计算机体系结构基础，熟悉CPU/GPU/NPU等处理器设计原理
熟悉深度学习算法及框架(如PyTorch)，有AI编译器或算子优化经验者优先
具备较强的编程能力(C/C++/Python)
有AI芯片相关研究或工作经验者优先
良好的问题分析和解决能力，具备较强的创新意识
优秀的沟通协作能力，能够在跨部门团队中高效工作