AI芯片引擎工程师

2023年11月17日 10:22    发布者:KT咨询
NO.680-【猎头职位:北京朝阳区/上海张江需要多位  AI芯片引擎工程师】联系人:edward-duan,邮箱:hr@kthr.com,微信也可查询职位了!打开手机微信,搜号码“KTHR_COM”或查找微信公众帐号“KT人才”或扫描以下二维码即可添加,欢迎大家关注!岗位职责: 1、负责开发AI推理框架,通过CPU/GPU/DSP/NPU的并行计算优化、架构设计、稀疏优化、异构调度等多种高性能优化技术打造业界领先的高性能异构AI推理引擎;2、负责AIGC、AI编译优化、端上训练等核心技术开发与优化;3、负责AI工具链开发及技术生态的建设,包括但不限于不同芯片的性能瓶颈分析与优化指引工具、模型图优化工具、可视化工具等。职位要求:1、硕士3年及以上学历,计算机/电子/信息/通信/自动化/软件等相关专业;2、精通C/C++,精通算法与数据结构,熟悉Python;3、熟练掌握NPU,CPU,GPU,DSP中的1个或多个平台高性能计算优化技术,深入理解计算机体系结构,熟悉并行计算优化、访存优化等;4、具备丰富的ARM NEON汇编优化或OpenCL/Metal/OpenGL/Vulkan/CUDA中的至少1种GPU优化经验,或熟练掌握WebAssembly、WebGL、WebGPU等技术,有这些技术的AI工程优化部署经验者优先;5、了解深度学习算法基本原理,熟悉神经网络基本架构和各算子计算方式,了解至少一种深度学习训练框架及其模型文件的解析,如Pytorch,Tensorflow;6、熟悉常用图像视觉计算库及深度学习推理计算库,例如arm compute library、tensorflow lite、ncnn、mnn、tnn、mkldnn、TensorRT、CoreML等,熟悉常用的深度学习算子优化方法例如Winograd, Strassen, Implicit GEMM等;7、掌握AI编译优化技术,例如TVM/MLIR/IREE等;8、了解主流 AIGC 算法模型原理,有AIGC模型加速优化经验者优先。