Ivyea Jobs 5948 roles · 69 companies · 刚刚
智元机器人 · 具身智能

AI框架开发工程师

北京 / 上海 社招 · 全职 工程开发

职位描述

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

任职要求

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