[招聘-百度-基础架构部]异构计算工程师
北京-异构计算工程师
工作职责:
- 参与AI推理引擎底层算子库的开发与优化,提升异构资源计算效率
- 参与大规模AI计算通信库及通信算法的开发与优化,提升大规模AI训练效率
- 探索面向CPU/GPU/FPGA/ASIC等多元化计算架构的编译系统开发、实现编译优化和算法加速
- 完成其他相关任务
- 热爱编程,精通C++/Python
- 具有独立开发能力,对AI算法和流框架有丰富的应用或开发经验
- 熟悉GPU/ARM/MIPS/DSP等任意异构计算平台
- 熟悉计算机体系结构,有汇编级别开发经验
- 有丰富的AI通信加速经验,量化/融合/拓扑优化
- 熟悉MPI,对不同网络拓扑结构的通信算法及底层通信函数有深入研究,对RDMA,GPU direct等技术有了解
PS:
校招,社招,不限欢迎投递,部门团队直面,提前批招聘,不影响后续22届校招 欢迎投递,大量HC
校招特别说明:校招生:基础扎实,有兴趣做AI底层计算加速与优化 即可
投递邮箱:1164028376@qq.com