跳转到主要内容

AI高性能算子开发工程师(实习)

Technology
上海奕行信息科技有限公司
上海市, 中国1个月前
全职

职位描述

该职位来源于猎聘 职位描述: 负责AI自研芯片高性能算子开发,基于自研硬件加速特性定制优化AI算法: 负责CNN/LLM等模型的性能验证优化,精度评估分析等; 参与AI算法模型在自研计算加速芯片上训练推理解决方案的实现 参与AI相关系统支持平台的开发和优化工作,提高通用性和易用性; 参与高性能算子开发的客户支持工作,推进客户平台适配及部署方案落地。 职位要求: 1.统招硕士及以上学历,计算机、通信、电子自动化,数学等相关专业优先 2.有扎实的C/C++/Python编程能力 3.熟悉至少一种深度学习软件框架,如Tensorflow/Caffe/PyTorch/MXNet/PaddlePaddle等, 4.熟悉大模型(如Deepseek/Llama/GPT/Qwen等)结构及其优化方法优先 5.熟悉主流计算机视觉任务,熟悉AI算法的训练推理和量化、蒸馏、剪枝等优化方法优先; 6.有cpu/dsp/gpu/npu等硬件架构下AI算子算法性能优化经验优先 7.有Risc-V或者CUDA并行计算开发经验优先 8.有一定AI算法基础,有相关科研成果(如相关论文、专利或者报告)优先; 9.有编程或者数学竞赛获奖,AI相关项目开发经验者优先;

Keywords
PyTorchApache MXNetCaffe

对这个职位感兴趣吗?