工作职责:
1、扩展AI编程框架,支持新的GPU类型;
2、面向新型加速器的体系结构特征,优化AI编译器的后端;
3、大规模分布式训练中多机计算、图执行、训练性能的优化等;
4、深度学习应用和框架的性能调优。
任职资格:
1、统招本科及以上学历,计算机专业或数学专业;有深度学习系统、图计算、高性能计算、编译器、计算机体系结构等工作背景;
2、熟悉常用AI编程框架中的一个,了解其结构和工作原理,如PyTroch、TensorFlow等;
3、有较强的动手能力,精通C++及Python编程语言;
4、工作经验3年以上,参与或负责过大型软件框架的设计和性能优化经验;
5、具有良好的团队合作意识;
6、工作作风积极、主动、勤思考。