AI加速软件工程师20K-40k

学历:硕士 | 工作年限:应届毕业生 | 年龄:35岁以下
最后刷新:2024/05/09 08:03:23

微信扫一扫:分享

微信里点“发现”,扫一下

立即沟通 申请职位
职位描述
职位介绍
岗位职责:
开发将 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.较强的沟通能力,领导能力和职业道德。
公司介绍

墨芯人工智能专注稀疏化计算,致力于提供云端和终端AI计算平台和服务。通过全球前沿的稀疏化技术优势,打造软硬协同的新一代智能计算平台。产品性能**行业,相较行业主流产品,在算力、能效比与成本等方面均实现数量级优化。面向互联网、运营商、金融、制造、医疗、交通、能源、生命科学、自动驾驶等众多行业与场景,提供高算力、低功耗、高性价比的AI算力服务,赋能前沿科技的进步与社会的智能化升级。

工作地点
上海-徐汇区徐汇区漕宝路650号1号楼1101
职位申请 快速申请 登录申请
确定 取消
芯片半导体职位来 高芯圈
登录 / 注册