岗位描述
工作内容
1.NPU芯片的AI编译器的设计与实现。针对SIMT,多级缓存,互联架构进行设计与实现多种关键优化;
2.机器人与具身智能场景的模型端到端性能优化,软硬件协同性能分析与设计;为端NPU引入前沿技术;
3.模型性能仿真等工具的设计与开发;
4.大模型推理引擎设计和开发,针对端NPU引入和改善关键优化,支持多存储层次MOE冷热分离优化等;
5.LLM、VLM相关压缩量化等工具链的开发;
任职条件
1.计算机、微电子、集成电路、电子信息、计算机体系结构等相关专业,本科及以上学历;
2.熟悉Linux平台上的C/C++、Python编程;理解常用数据结构和算法;
3.具备编译器 + LLM推理 + 软硬件协同探索方向的学习基础或项目经历,对该领域的系统性问题有浓厚的探索欲;
4.具备良好的学习能力、问题拆解能力和团队协作意识;
加分项:
1.有AI芯片编译器、量化相关的项目和实习经验;
2.了解计算机体系结构及相关量化分析方法;有软硬件协同设计与分析项目经验优先;
3.有GPGPU相关编译与算子优化经验,有NPU与DSA优化经验优先;
4.熟悉异构编程环境的相关研发;
5.有MLIR/TVM/TileLang/Triton/vLLM等社区开发经验者优先。