上海天数智芯半导体有限公司
上海
计算框架研发工程师
20-40K·15薪
南京
3-5年
本科
全职
职位描述
岗位职责：
1.基于自研GPGPU适配/优化/维护常用AI计算框架TensorFlow、PyTorch、MXNet等；
2.实现易扩展、高性能、大规模的AI模型训练系统，发挥自研GPGPU的优越性；
任职要求：
1.具有计算机相关领域背景，熟悉AI领域更好；
2.熟练使用Python/C/C++编程；
3.熟悉CUDA编程。
4.对TensorFlow/PyTorch实现原理熟悉、熟悉底层代码的实现。
5.对分布式计算/训练框架有开发经验。
6.熟悉TensorRT/TVM等推理引擎，了解原理。
1.基于自研GPGPU适配/优化/维护常用AI计算框架TensorFlow、PyTorch、MXNet等；
2.实现易扩展、高性能、大规模的AI模型训练系统，发挥自研GPGPU的优越性；
任职要求：
1.具有计算机相关领域背景，熟悉AI领域更好；
2.熟练使用Python/C/C++编程；
3.熟悉CUDA编程。
4.对TensorFlow/PyTorch实现原理熟悉、熟悉底层代码的实现。
5.对分布式计算/训练框架有开发经验。
6.熟悉TensorRT/TVM等推理引擎，了解原理。
公司介绍
上海天数智芯半导体有限公司（简称“天数智芯”）于2018年正式启动GPGPU大芯片设计，是中国第一家GPGPU高端大芯片及超级算力提供商。
公司以“成为智能社会的赋能者”为使命，专注于云端服务器级的通用高性能计算大芯片，瞄准以云计算、人工智能、数字化转型为代表的数据驱动技术市场，解决核心算力瓶颈问题。重点打造自主可控、国际一流的通用、标准、高性能云端计算GPGPU大芯片，从芯片端解决算力问题；并推出面向5G应用需求的云端推理芯片，与进口主流GPGPU生态兼容，并提供国产化选项，意在将超级计算发展为水电煤一样的基础设施，以更可信、更高效、更经济的方式帮助各行业实现算力升级。
公司以“成为智能社会的赋能者”为使命，专注于云端服务器级的通用高性能计算大芯片，瞄准以云计算、人工智能、数字化转型为代表的数据驱动技术市场，解决核心算力瓶颈问题。重点打造自主可控、国际一流的通用、标准、高性能云端计算GPGPU大芯片，从芯片端解决算力问题；并推出面向5G应用需求的云端推理芯片，与进口主流GPGPU生态兼容，并提供国产化选项，意在将超级计算发展为水电煤一样的基础设施，以更可信、更高效、更经济的方式帮助各行业实现算力升级。
南京建邺区新城科技园国际研发总部园2栋
面试经验
相似职位
