岗位描述
角色和团队介绍:
团队致力于构建全国产化、高性能的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系统实战经验
地点:杭州、上海