AI推理框架工程师
岗位摘要
针对NLP、语音、CV等任务的训练性能优化；针对NLP、语音、CV等任务的推理性能优化；架构设计、评审、优化、技术攻坚；神经网络算子开发；推理、训练框架维护工作；计算机相关专业本科以上学历，一年以上开发工作经验
岗位职责
- 针对NLP、语音、CV等任务的训练性能优化
- 针对NLP、语音、CV等任务的推理性能优化
- 架构设计、评审、优化、技术攻坚
- 神经网络算子开发
- 推理、训练框架维护工作
任职要求
- 计算机相关专业本科以上学历，一年以上开发工作经验
- 熟悉PyTorch、TensorFlow等主流深度学习工具及其演进和发展
- 熟悉AI框架和异构并行计算，对GPU/OpenCL/CUDA/OpenMP有深刻理解
- 丰富的软件和算法架构设计能力，精通深度学习、强化学习、运筹优化、分布式计算等软件架构
- 主导或者参与过关键领域的架构和软件系统设计、开发交付等工作
工作地址
北京海淀区蓟门壹号8层