职位详情

职位描述: 作为机器学习系统工程师,您将负责开发和优化高效的 AI 系统,确保我们的机器学习模型在各种硬件平台上,如 CUDA、Metal、昇腾等 GPU 上,达到最佳性能。您将与我们的 AI 研发团队紧密合作,推动机器学习技术的创新和应用。 主要职责: • 设计和实现高效的机器学习算法和算子,优化在 AI 硬件上的执行性能; • 使用 CUDA、Metal 以及 CANN 进行深度学习模型的底层性能优化; • 分析和改进机器学习工作流,提高数据处理和模型训练的效率; • 与软件工程师和 AI 科学家合作,协同提升产品整体性能; • 跟踪最新的机器学习和系统优化技术,持续提升系统性能; • 解决复杂的技术问题,确保系统的稳定性和可扩展性。 职位要求: • 拥 985 或 211 工程院校毕业,计算机科学或相关领域的本科及以上学历; • 熟悉深度学习和机器学习算法,有 3 年以上工程经验; • 精通 CUDA 编程和 NVIDIA GPU 架构,或熟悉 Metal 框架和 Apple GPU 架构; • 强大的 C/C++编程能力,熟悉 Python 或其他脚本语言; • 有使用 TensorFlow、PyTorch 或其他深度学习框架的经验优先; • 具备优秀的分析和问题解决能力,能够处理复杂的技术挑战; • 具备创新性思维,能够仰望星空,但同时做事坚实可靠、脚踏实地; • 具备良好的团队协作精神,是友善、稳定、可信赖的伙伴; • 具备优秀的道德品质和价值观,能够在工作中体现诚信和责任感。 加分项: • 公开发布的代表性作品或开源项目; • 在业界知名企业工作经验; • 优秀的英语沟通能力。

工作地址

北京望京科技园A座6楼

公司信息

相似的职位

机器学习系统工程师 (Machine Learning Systems Engineer)北京NaN机器学习40-70k12薪

上海先枫信息科技有限公司

机器人导航算法工程师

,

深度学习与 3D 算法专家

,

数据标注