Ivyea Jobs 8806 roles · 130 companies · 刚刚
Momenta · 具身智能

CUDA性能优化研发工程师

北京 / 苏州 / 上海 社招 · 全职 推理 / 部署优化

职位描述

研发深度学习推理高性能计算库 优化CNN/Transformer模型在CUDA平台上的推理效率

任职要求

计算机/数学/电子信息等相关专业毕业,硕士以上学历。 熟悉CUDA/C++开发,熟悉计算机体系结构,精通高性能计算,2年以上开发经验。 熟悉Onnx/Pytorch/TVM等深度学习计算库/框架 对计算性能有极致追求 ACM/ICPC/NOI/ASC等竞赛获奖者优先。
Momenta去官方页面投递 →