1.负责公司C++并行算法的GPU(CUDA)加速与移植。
2.负责公司深度学习架构的设计与维护pytorch(libtorch)、tensorRT。
3.负责公司算法库(GPU版本)在Window及Linux下的更新和维护。
任职要求:
1.熟练掌握GPU(CUDA)编程,并能将CPU并行算法进行CUDA加速;
2.有良好的数据结构与算法功底,熟练掌握 C,C++及Python;
3.有pytorch(libtorch),tensorflow等机器学习平台使用经验的优先;
4.具有丰富的编程经验,能独立完成代码重构、移植、诊断调试与性能调优等,具有Linux系统下程序开发的经验;
5.具备良好的问题定位分析能力,有良好的沟通能力和团队合作能力,善于沟通,工作自主驱动。