中国电子科技南湖研究院(简称“南湖院”)是中国电子科技集团有限公司和浙江省嘉兴市政府共同建立的新一代技术创新载体。南湖院AI框架团队,依托中电海康丰富AI产业资源,从芯片、框架、算法多个维度深入布局,对标海外AI训练明星级产品,致力打造国有自主的AI训练软硬件基础设施,解决国家卡脖子的难题,为中国社会提供AI算力保障,最终孵化一家该领域高新技术上市公司。
岗位职责:
1. 框架设计,负责主流机器学习框架对比分析,根据业务需求与自研芯片特色设计AI框架模块,保障框架整体功能;
2. 算子开发,负责深度学习、科学计算核心算子的优化开发,以及计算图优化、即时编译相关的开发工作;
3. 芯片适配,负责与芯片团队一起,完成AI框架与自研芯片的适配工作,并完成商业落地验证。
任职要求:
1)熟练掌握至少一种编程语言,如Python、C、C++等,具备专业编程实现能力者优先;
2)熟悉CUDA编程、多卡通信原理、计算图编译者优先;
3)有机器学习基础,了解神经网络常用算子,可以根据不同任务优化底层算子逻辑。
4)熟悉现有主流深度学习框架(如MindSpore、Paddle Paddle,TensorFlow、PyTorch等),对其中一款有深入理解,并能分析其优缺点者优先;
5)有模型加速与性能优化项目经验者优先;
6)积极乐观,有创业精神者优先;
7)工作地点:北京 或 嘉兴。