墨芯人工智能

AI加速软件工程师  

不限深圳 上海 北京经验经验不限本科校招

年底双薪 岗位晋升 弹性工作 绩效奖金 带薪年假 股票期权 交通补助 定期体检 五险一金

投递简历 编辑个人简历

 

岗位职责

开发将 Tensorflow / Pytorch / MxNet 等 AI编程框架模型布署至 AI加速计算平台的模型编译器和运行库,一些工作职责包括:

  1. 开发转化、融合、拆解算子、生成指令、以及分配资源的AI模型编译器;
  2. 了解计算平台的指令集和体系结构,开发平台上的算子、并设计内存分配、执行顺序和分区的优化策略; 
  3. 利用通用运算平台 (CPU/GPU) 开发算子/模型的运行库,模拟其在AI加速平台上的精度和性能; 
  4. 制定测试计划和测试框架,以验证软件工具链的正确性和有效性。

任职要求

  1. 计算机科学或相关领域的本科/硕士/博士学位或同等学历;
  2. 熟练使用C / C ++ 编程。在AI模型开发框架(如Tensorflow / PyTorch / MxNet)方面有经验;
  3. 了解常见AI模型算子的概念。具备类似 TensorRT / TVM / TF-Lite / MNN / NCNN / ARM-NN 等 AI 推理框架的经验者为佳;
  4. 具备 TVM / XLA / Glow / NGraph之类的 AI 编译器方面的经验者为佳;
  5. 具有优化数值计算库(例如CUDNN / MKL)经验者为佳;
  6. 熟悉视频编解码器、DSP、PCIe协定、嵌入式开发、或CPU,GPU 及 AI ASIC架构者为佳;
  7. 较强的沟通能力,领导能力和职业道德。
分享:

公司基本信息

所属领域:智能硬件

公司规模:50-150人

工作地点:深圳 上海 北京 硅谷

融资情况:A轮