返回岗位列表

壁仞科技

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

地点:上海 薪资:薪资未公开 日期:2026-01-29

岗位摘要

基于特定硬件与深度学习框架实现高效的深度学习算法和模型;优化现有的深度学习框架及分布式训练框架实现,提高计算效率和资源利用率;参与深度学习框架的架构设计和代码审查;与用户(算法工程师)紧密合作,理解他们的需求并提供技术支持

岗位职责

  • 基于特定硬件与深度学习框架实现高效的深度学习算法和模型
  • 优化现有的深度学习框架及分布式训练框架实现,提高计算效率和资源利用率
  • 参与深度学习框架的架构设计和代码审查
  • 与用户(算法工程师)紧密合作,理解他们的需求并提供技术支持
  • 跟踪最新的深度学习研究进展,并将其应用到实际项目中
  • 编写技术文档,包括设计说明、用户手册和API文档
  • 为团队成员提供深度学习框架的培训和指导
  • 参与跨部门项目,提供深度学习解决方案

任职要求

  • 计算机科学、电子工程或相关领域的硕士学位
  • 至少2年在深度学习框架开发和维护方面的经验
  • 精通至少一种深度学习框架(如TensorFlow, PyTorch, Caffe等)
  • 熟练掌握Python、C++编程语言
  • 熟悉Linux操作系统和Shell脚本编程
  • 了解并行计算和GPU编程,熟悉CUDA, OpenCL,OpenMP异构编程经验的优先
  • 具备良好的数学和算法基础
  • 优秀的问题解决能力和创新思维
  • 英语读写能力良好