返回岗位列表

OpenAI

软件工程师(编译器/内核方向)

地点:美国 薪资:薪资未公开 日期:2021-09-21

岗位摘要

设计和优化核心机器学习系统;编写高可靠性的底层代码;推进为模型提供动力的算法和基础设施;设计和构建编译器、语言以及高性能内核;推进Triton及其后端,开发新的编译器通道;创建用于为全新硬件编写快速、正确且深度优化的内核所需的工具

岗位职责

  • 设计和优化核心机器学习系统
  • 编写高可靠性的底层代码
  • 推进为模型提供动力的算法和基础设施
  • 设计和构建编译器、语言以及高性能内核
  • 推进Triton及其后端,开发新的编译器通道
  • 创建用于为全新硬件编写快速、正确且深度优化的内核所需的工具
  • 与硬件团队紧密合作,解锁新功能并确保定制芯片能够支持下一代前沿模型

任职要求

  • 3年以上相关工程经验,最好在系统、编译器、机器学习框架或性能工程领域
  • 具备端到端解决问题的能力,包括根据需要学习新的硬件和软件领域
  • 加分项:对机器学习框架(如PyTorch、JAX、TensorFlow)或编译器(如LLVM、MLIR、XLA)有贡献,或有编程定制或专用加速器的经验

加分项

  • 对机器学习框架(如PyTorch、JAX、TensorFlow)或编译器(如LLVM、MLIR、XLA)有贡献,或有编程定制或专用加速器的经验