职位详情
资深模型推理优化工程师
2.5-5万
北京视知觉智能科技有限公司
北京
3-5年
硕士
03-19
工作地址

北京市留学人员海淀创业园305 室

职位描述
岗位职责
1. 负责智能超声诊疗模型在多种不同硬件平台的极致性能优化,结合编译优化、并行计算优化、图融合、高效 CUDA 算子开发实现行业领先的超声设备端推理性能。
2. 针对特定NPU硬件计算平台,基于对硬件体系结构的深入理解,实现对硬件的高效利用。
3. 针对Pytorch/CUDA相关GPU计算任务,进行算子和系统优化,提升训练/推理效率。

岗位要求
1. 工作3年以上,计算机/数学/物理/电子工程/自动控制专业硕士及以上学历。
2. 优秀的代码能力,熟练掌握C/C++或Python,有 CUDA 开发经验,熟悉TVM/TensorRT/Triton/Cutlass等推理框架优先。
3. 熟悉计算机体系结构,对GPU、NPU等有比较深刻的理解。
4. 熟悉CNN/Transformers/DETR等主流模型原理。5. 良好的团队协作能力;创新性强,有良好的动手实现能力,对技术有热情。

以担保或任何理由索取财物,扣押证照,均涉嫌违法,请提高警惕

立即申请