计算框架研发工程师
10K-15K
收藏职位 申请职位
江苏省| 人数: 若干| 经验:不限| 性别:不限| 年龄:不限| 学历:不限| 0人浏览
温馨提示: 求职中如遇招聘方扣押证件、要求提供担保或收取财物、强迫入股或集资、收取不正当利益或其他违法情形,请立即举报 ;如遇岗位要求海外工作,请提高警惕,谨防诈骗
上海天数智芯半导体有限公司
职位描述
岗位职责:
1.基于自研GPGPU适配/优化/维护常用AI计算框架TensorFlow、PyTorch、MXNet等;
2.实现易扩展、高性能、大规模的AI模型训练系统,发挥自研GPGPU的优越性;

任职要求:
1.具有计算机相关领域背景,熟悉AI领域更好;
2.熟练使用Python/C/C++编程;
3.熟悉CUDA编程。
4.对TensorFlow/PyTorch实现原理熟悉、熟悉底层代码的实现。
5.对分布式计算/训练框架有开发经验。
6.熟悉TensorRT/TVM等推理引擎,了解原理。
工作地点
南京建邺区新城科技园国际研发总部园2栋