Ivyea Jobs 8388 roles · 122 companies · 刚刚
Momenta · 具身智能

模型训练优化开发工程师

北京 / 深圳 / 苏州 / 上海 社招 · 全职 预训练 / 基座模型

职位描述

岗位职责 1. 负责多种不同硬件平台的极致性能优化,结合编译优化、并行计算优化、图融合、高效 CUDA 算子开发实现行业领先的车端推理性能。 2. 针对特定NPU硬件计算平台,基于对硬件体系结构的深入理解,实现对硬件的高效利用。 3. 针对Pytorch/CUDA相关GPU计算任务,进行算子和系统优化,提升训练/推理效率。 4.负责优化训练框架,尽可能高效发挥硬件和系统软件性能等

任职要求

岗位要求 1. 工作2年以上,计算机/数学/物理/电子工程/自动控制专业本科及以上学历优先。 2. 优秀的代码能力,熟练掌握C/C++或Python 3. 熟悉PyTorch/TensorFlow等训练框架的原理、软件架构。 4. 熟悉CNN/Transformers/DETR等主流模型原理。 5. 良好的团队协作能力;创新性强,有良好的动手实现能力,对技术有热情。
Momenta去官方页面投递 →