Job Description

AI框架开发工程师

北京、上海

全职

职位描述

参与AI 框架相关的软件栈以及运行时的软件架构设计及产品实现，打通从Framework到驱动界面的流程，实现高性能的推理和训练解决方案； 支持自研训推一体框架的开发，并在训练速度、推理性能、生态兼容程度方面达到业界领先； 降低软件方案的应用复杂度，实现高可用的部署和优化流程，实现下游用户短时间内完成产品级的开发环境集成； 维护软件栈的开发复杂度和产品质量，参与软硬件联合优化讨论，确保前期硬件功能、性能特性落地，实现系统级的垂直优化； 参与完整的产品周期开发，包括但不限于算法开发，算法评测、算法集成发布、模型优化编译、算法交付部署等，并支持框架维护升级。

职位要求

计算机/电子工程专业、硕士及以上（博士优先） 5年以上产品级的AI框架开发和维护经验，熟悉PyTorch/Jax等主流AI框架的代码结构、关键特性及实现 核心技术: 熟练掌握Transformer架构的原理及基于Transformer架构模型的计算、访存特征； 熟练掌握常见强化学习、监督学习等算法训练流程，并主导过算法生产到交付全流程； 熟练掌握计算图编译和优化的基本理论，了解PyTorch/TensorFlow中图相关的优化流程，及硬件架构相关的优化处理方法； 掌握典型的集合通信接口，对分布式框架中的通信后端有一定的了解； 熟悉常见的端侧AI框架如MNN、NCNN设计与架构； 精通C++/C, Python编程，精通数据结构和算法的基础知识，具有良好的架构能力和代码质量意识。 有一定的项目管理经验和知识储备，良好的工作习惯和沟通表达能力 关键加分项: 基于PyTorch/TensorFlow主导过完整的算法训练到部署的落地流程 熟悉具身领域常用生态框架，如lerobot、gr00t dream pipeline等 熟悉大语言模型部署的性能关键点 分布式推理/训练系统的设计开发经验 有创新思维意识和优秀的逻辑分析能力，能够结合既往项目经验和具体问题并给出合理的分析和解答。

投递