【2027校招】AI框架/算子开发工程师

角色和团队介绍：

团队致力于构建全国产化、高性能的AI基础设施软件平台，实现对行业主流框架的全面支持。在这里，你将与硬件和软件同事共同协作，参与高性能算子的开发与优化，国产AI芯片与主流AI框架的生态对接，并支持前沿AI应用方案的开发与系统调优。欢迎加入我们，与我们携手前行，共同成长。

岗位职责：

1.参与深度学习框架（PyTorch等）的核心模块开发、性能优化与功能扩展

2.负责高性能算子设计与实现与调优

3.负责大规模训练与推理场景的系统优化，解决性能、内存与通信瓶颈

4.探索前沿技术方向（如编译优化、自动并行、异构计算等）并推动落地

岗位要求：

1.2027届校招应届生

2.计算机、电子工程、人工智能等相关专业本科及以上学历

3.熟练掌握 c++, python, bash 等编程和脚本语言

4.理解AI框架应用及常见的云端AI模型(如cv、transformer类)

5.熟练掌握pytorch的使用

6.有上进心、责任心、探索精神，出色的团队合作和交流能力

7.具备英文技术文档阅读能力

* 加分项：

1. 有高性能计算与算子优化（如Kernel开发、性能剖析与调优）经验

2.了解 cuda等并行计算架构优化方法，对并行计算感兴趣

3.有参与开源框架开发(如vllm/sglang等)、异构计算或大规模AI系统实战经验

地点：杭州、上海