职位描述
角色和团队介绍:
团队致力于构建全国产化、高性能的AI基础设施软件平台,实现对行业主流框架的全面支持。在这里,你将与硬件和软件同事共同协作,参与高性能算子的开发与优化,国产AI芯片与主流AI框架的生态对接,并支持前沿AI应用方案的开发与系统调优。欢迎加入我们,与我们携手前行,共同成长。
岗位职责:
1. 参与深度学习框架(PyTorch等)的核心模块开发、性能优化与功能扩展;
2. 负责高性能算子设计与实现与调优;
3. 负责大规模训练与推理场景的系统优化,解决性能、内存与通信瓶颈;
4. 探索前沿技术方向(如编译优化、自动并行、异构计算等)并推动落地。
岗位要求:
1. 2026届校招应届生,
2. 计算机、电子工程、人工智能等相关专业本科及以上学历
3. 熟练掌握 c++, python, bash 等编程和脚本语言;
4. 理解AI框架应用及常见的云端AI模型(如cv、transformer类);
5. 熟练掌握pytorch的使用
6. 有上进心、责任心、探索精神,出色的团队合作和交流能力;
7. 具备英文技术文档阅读能力;
* 加分项:
1. 有高性能计算与算子优化(如Kernel开发、性能剖析与调优)经验
2. 了解 cuda等并行计算架构优化方法,对并行计算感兴趣;
3. 有参与开源框架开发(如vllm/sglang等)、异构计算或大规模AI系统实战经验






-





