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