AI Infra工程师

一、AI算子技术 1、参与大模型训练推理算子的设计与开发，涵盖大模型训练、推理、数据合成等关键业务场景，实现算子加速，推动成果落地产品并产生商业价值； 2、参与算子性能优化、精度分析和计算图优化等方案的开发，通过数学等价变换、流水编排、深度融合、调度优化、Tile优化等手段提升算子性能，为端到端训推系统性能提供基础支撑； 3、参与高性能算子库、基础加速库、算子开发框架的编码与测试 4、参与自动算子生成工具的研发，利用AI Agent技术提升算子开发的自动化程度和效率； 岗位要求 1、人工智能、机器学习、计算数学、应用数学、软件工程、计算机科学、体系结构、统计数学、数据科学等相关专业本科或硕士学历； 2、具备较强的代码编写和算法实现能力，了解大模型架构、AI框架、AI系统、分布式计算与存储系统、并行软件开发； 3、具备较强的学习能力、自驱意识、团队意识和责任意识，能够积极参与团队协作，快速适应技术迭代； 4、具有大模型训推系统、算子开发调测实践经验者优先，掌握算子优化和性能调测的常用方法和工具（如Profiling工具、Tiling策略等）者优先； 5、有相关实习经验或高水平数学/计算机竞赛（如ACM/ICPC、各类数学竞赛）获奖经验者优先；在领域顶级会议发表过文章者更佳。 二、AI训推系统 1、参与Agent、语言、多模态、生成式推荐、空间智能等大模型训练和推理的关键技术与系统架构的开发与实现，协助完成算法突破，并将成果落地产品，产生商业价值； 2、参与训推框架和加速特性的设计与开发，包括预训练、RL后训练、Agentic AI、推理引擎和服务化、AI训推技术辅助编程等模块的编码与调试； 3、参与AI训练/推理系统的优化方案设计和开发，在计算、内存、通信、IO等方面开展软硬协同优化，提升系统性能； 4、跟踪业界和学术界AI计算训推算法、训练系统优化、推理系统优化领域的最新进展，参与业界顶会，参与开源社区贡献，学习并引入先进技术，持续提升系统竞争力。 岗位要求： 1、人工智能、机器学习、软件工程、计算机科学、体系结构、统计数学、数据科学等相关专业本科或硕士学历； 2、具备较强的代码编写和算法实现能力，了解大模型架构、AI框架、AI系统、分布式计算与存储系统、并行软件开发； 3、具备较强的学习能力、自驱意识、团队意识和责任意识，能够积极参与团队协作，快速适应技术迭代； 4、有相关实习经验或高水平竞赛（如ACM）获奖经验者优先；在领域顶级会议发表过文章者更佳。