部门
AI中心
岗位名称
AI软件系统工程师
岗位职责及岗位要求
岗位职责:1.研究面向大模型和AI芯片(如GPU或下一代AI芯片)的软件栈技术包括但不限于大模型算子、AI编译器、推理框架、分布式框架、量化方法、强化学习框架等。2.开发相应的软件项目,通过开源或社区合作的方式产生工业级影响力。3.发表顶级学术论文或申请专利。岗位要求:(一)基本要求:1.硕士学历需具备2年及以上AI软件相关工作经历2.熟练掌握C/C++/Python等至少一门编程技能,有扎实的编程基础3.了解至少一款GPU或AI芯片架构,并有关相关编程经验4.对大模型或其他AI模型的计算负载有基本了解(二)其他要求(有一下任意一项经验者优先考虑):1.在AI系统领域发表过顶级学术论文;2.熟悉GPU或其他AI芯片上算子编程与优化(如CUDA,Cutlass,Cute,Triton,TileLang等);3.熟悉一项或多项AI软件栈技术或框架:如Torch.compiler,Triton,TileLang,vLLM,SGLang,MLIR,TVM,XLA,Pytorch,TensorFlow,ONNX等;4.熟悉主流模型架构及相关算子优化技术(如FlashAttention、FlashMLA、DeepEP等);5.熟悉NCCL或类似通信编程技术,或者有过分布式和异步编程经验;6.有超算或并行编程经验(如熟悉分布式矩阵乘法、MPI等);7.有其它AI芯片的算子开发、模型优化、低精度量化、部署经验者。
简历筛选人邮箱
xuejilong@ime.ac.cn