AI框架适配优化研发工程师

飞腾信息技术有限公司 长沙市 社会职位 全职
关 注 投 递

职位信息

  • 工作地址: 长沙开福区CFC富兴时代富兴时代T2栋
  • 招聘人数: 若干 人
  • 工作性质: 全职
  • 薪资:15000元 - 40000元 (月薪)

职位描述

岗位职责：
1. 基于计算设备编程框架，开发与主流深度学习框架兼容的算子
2. 基于计算设备底层软件栈，对主流深度学习框架进行适配开发
3. 分析主流深度学习框架在计算设备上的性能瓶颈，并进行性能优化
4. 主流大模型并行训练框架适配与优化
任职要求：
1. 熟悉掌握深度学习算法，能熟练地训练神经网络模型
2. 熟悉主流框架中的常用算子及其数学意义，有实现经验的优先
3. 熟悉主流深度学习框架，包括但不限于Pytorch、Tensorflow、Jax、PaddlePaddle、MindSpore等，对其中之一或多个的底层实现有较深入研究的优先
4. 熟悉并行训练框架，了解大模型训练或微调技术，有实际项目经验的优先
5. 熟悉CUDA、OpenCL等异构编程框架的优先；熟悉CUDA软件栈、oneAPI软件栈或ROCm软件栈的优先
6. 熟练掌握Python和C/C++，了解Python与C/C++相互调用方法；熟悉Linux，熟练掌握Shell

职位要求

  • 工作年限: 3-5
  • 最低学历: 本科
  • 年龄要求: 无

公司信息

  • 飞腾信息技术有限公司
  • 企业类型: 企业
  • 企业业务: 研究开发、技术咨询与服务
  • 企业网址: http://www.phytium.com.cn/
  • 企业简介:

    飞腾信息技术有限公司是一家快速成长中的中国芯片设计企业，总部位于天津滨海高新技术产业开发区，在北京、广州等地均设有运营、销售和研发中心。主要致力于高性能、低功耗集成电路芯片的设计、生产、销售与服务，为用户提供安全可靠、高性能、低功耗的CPU、ASIC、SoC等芯片产品、

推荐职位

+ 查看更多

合规专员/助理

飞腾信息技术有限公司 长沙市 社会职位 全职
查看详情

更新时间: 2024/09/29

合规经理/主管

飞腾信息技术有限公司 长沙市 社会职位 全职
查看详情

更新时间: 2024/09/29

POC适配验证高级工程师

飞腾信息技术有限公司 北京市 社会职位 全职
查看详情

更新时间: 2024/09/29

生态合作项目PMO

飞腾信息技术有限公司 北京市 社会职位 全职
查看详情

更新时间: 2024/09/29

SRAM电路设计工程师

飞腾信息技术有限公司 上海市 社会职位 全职
查看详情

更新时间: 2024/09/29