工作职责:-负责深度学习框架的研发工作,增强框架完备性,实现框架基础功能以满足各类深度学习任务的需求-负责深度学习框架的API研发工作,提高API性能、易用性等-负责深度学习框架的性能优化工作,包括但不限于功能模块在CPU/GPU上的深度优化工作-参与深度学习框架的易用性优化工作,使开发者能够以更简单的方式实现各类任务,降低学习成本和开发成本-参与深度学习框架编译器的研发工作,增强编译器能力,支撑框架优化模型性能任职资格:-有较强的工程实现能力,对于给定算法能够快速实现并能对时间、空间进行优化-精通C/C++编程,熟悉Python编程-熟练掌握各种数据结构和算法,了解矩阵计算相关算法-有较强的学习能力和沟通能力,具备良好的团队合作精神-有CUDA高性能编程开发经验-对AI编译器有一定了解,或者有相关的开发经验-了解深度学习模型,对机器学习有一定基础-有深度学习框架使用或开发经验,包括但不限于PaddlePaddle,TensorFlow, PyTorch等
上班地址:-北京-北京市/北京市/海淀区百度科技园西北旺东路10号院
该职位发布已超过90天,可能已过期!