上海 [切换城市] 上海招聘

计算框架研发

上海天数智芯半导体有限公司

  • 公司规模:150-500人
  • 公司性质:民营公司
  • 公司行业:计算机软件

职位信息

  • 发布日期:2020-11-20
  • 工作地点:上海-浦东新区
  • 招聘人数:1人
  • 工作经验:2年经验
  • 学历要求:本科
  • 职位月薪:2.5-3万/月
  • 职位类别:算法工程师

职位描述

1 工作内容

 

a. 基于自研GPGPU适配/优化/维护常用AI计算框架TensorFlowPyTorchMXNet等。

b. 实现易扩展、高性能、大规模的AI模型训练系统,发挥自研GPGPU的优越性。

c. 基于自研芯片设计构建高性能推理引擎。

 

2 资历要求

 

a. 计算机硕士+1年相关工作经验,或计算机学士+3年工作经验

b. 具有计算机相关领域背景,熟悉AI领域更好。

c. 熟练使用Python/C/C++编程。

d. 熟悉CUDA编程。

e. TensorFlow/PyTorch实现原理熟悉、熟悉底层代码的实现。

f. 对分布式计算/训练框架有开发经验。

g. 熟悉TensorRT/TVM等推理引擎,了解原理。

h. 有良好的的团队合作精神和沟通能力


职能类别:算法工程师

关键字:TensorFlowgpuPyTorchMXNet

公司介绍

上海天数智芯半导体有限公司(简称“天数智芯”)于2018年正式启动GPGPU大芯片设计,是中国一家GPGPU高端大芯片及超级算力提供商。
 
公司以“成为智能社会的赋能者”为使命,专注于云端服务器级的通用高性能计算大芯片,瞄准以云计算、人工智能、数字化转型为代表的数据驱动技术市场,解决核心算力瓶颈问题。重点打造自主可控、国际一流的通用、标准、高性能云端计算GPGPU大芯片,从芯片端解决算力问题;并推出面向5G应用需求的云端推理芯片,与进口主流GPGPU生态兼容,并提供国产化选项,意在将超级计算发展为水电煤一样的基础设施,以更可信、更高效、更经济的方式帮助各行业实现算力升级。

联系方式

  • 公司地址:地址:span自由贸易实验区亮秀路112号Y2座202、203和204室