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

职位概述:

作为深度学习框架工程师(性能优化),您将加入我们的研发团队,参与深度学习框架的开发和维护工作。您将负责前沿的大语言模型算法等深度学习算法在特定硬件上的实现、优化与应用交付。

主要职责:

基于特定硬件与深度学习框架实现高效的深度学习算法和模型。

优化现有的深度学习框架及分布式训练框架实现,提高计算效率和资源利用率。

参与深度学习框架的架构设计和代码审查。

与用户(算法工程师)紧密合作,理解他们的需求并提供技术支持。

跟踪最新的深度学习研究进展,并将其应用到实际项目中。

编写技术文档,包括设计说明、用户手册和API文档。

为团队成员提供深度学习框架的培训和指导。

参与跨部门项目,提供深度学习解决方案。

任职要求:

计算机科学、电子工程或相关领域的硕士学位。

熟悉PyTorch 训练框架。

熟练掌握Python、C++编程语言。

熟悉Linux操作系统和Shell脚本编程。

具有行业前沿的自然语言、视觉、语音或其他垂直领域的深度学习模型训练、推理部署经验。

了解并行计算和GPU编程,熟悉CUDA, OpenCL,OpenMP异构编程经验的优先

具备良好的数学和算法基础。

优秀的问题解决能力和创新思维。

良好的团队合作精神和沟通能力。

英语读写能力良好。

加分项:

有在开源项目中贡献代码的经验。

有发表深度学习相关学术论文的经历。

对机器学习、计算机视觉或自然语言处理有深入理解。

公司地点:上海闵行区壁仞科技16号楼

公司简介:

通用智能计算公司,国产高端通用智能计算芯片,提供软硬件解决方案;应用于人工智能、云计算、图形渲染、大数据处理等新兴领域。

职位发布者:苏先生

上海壁仞科技股份有限公司

融资阶段:

公司规模:500~999人

相似职位: