深度学习框架工程师(性能优化)

薪资面议
本科
上海市闵行区陈行公路2388号浦江科技广场16号楼M层
2025-04-21










职位概述:
作为深度学习框架工程师(性能优化),您将加入我们的研发团队,参与深度学习框架的开发和维护工作。您将负责前沿的大语言模型算法等深度学习算法在特定硬件上的实现、优化与应用交付。
主要职责
基于特定硬件与深度学习框架实现高效的深度学习算法和模型。
优化现有的深度学习框架及分布式训练框架实现,提高计算效率和资源利用率。
参与深度学习框架的架构设计和代码审查。
与用户(算法工程师)紧密合作,理解他们的需求并提供技术支持。
跟踪最新的深度学习研究进展,并将其应用到实际项目中。
编写技术文档,包括设计说明、用户手册和API文档。
为团队成员提供深度学习框架的培训和指导。
参与跨部门项目,提供深度学习解决方案。
任职要求
计算机科学、电子工程或相关领域的硕士学位。
至少2年在深度学习框架开发和维护方面的经验。
精通至少一种深度学习框架(如TensorFlow, PyTorch, Caffe等)。
熟练掌握Python、C++编程语言。
熟悉Linux操作系统和Shell脚本编程。
了解并行计算和GPU编程,熟悉CUDA, OpenCL,OpenMP异构编程经验的优先
具备良好的数学和算法基础。
优秀的问题解决能力和创新思维。
良好的团队合作精神和沟通能力。
英语读写能力良好。
加分项:
有在开源项目中贡献代码的经验。
有发表深度学习相关学术论文的经历。
对机器学习、计算机视觉或自然语言处理有深入理解。