AI框架开发工程师

AI框架开发工程师

工作地点：上海/杭州

岗位职责

参与 AI 芯片深度学习框架的适配、开发与优化，支持主流 AI 模型在芯片上的训练和推理；

参与 PyTorch 等主流深度学习框架核心模块的开发与适配，解决模型运行过程中的功能、精度及性能问题；

参与计算图、算子调度、内存管理等框架核心能力的设计与优化；

针对 AI 模型进行性能分析和瓶颈定位，与算子、编译器及 Runtime 团队协作进行性能优化；

参与 AI 软件栈及框架生态建设，持续提升框架的易用性、稳定性和性能。

任职要求

本科及以上学历，计算机、软件工程、电子信息、数学等相关专业；

具备扎实的数据结构、算法、操作系统及计算机体系结构基础，熟悉 C / C++ 和 Python；

熟悉 PyTorch 等至少一种主流深度学习框架，了解其基本架构和运行机制；

了解深度学习模型训练或推理基本流程，对计算图、算子、自动微分、内存管理等相关技术有一定理解；

了解 GPU / NPU 等 AI 加速器及并行计算基本原理，有 CUDA 等相关开发经验者优先；

具备较强的逻辑分析、Debug 和问题解决能力，有较强的学习能力、自驱力和技术钻研精神。

加分项

有 PyTorch / TensorFlow 等框架开发、分布式训练、模型部署或开源项目贡献等相关经历者优先。