返回岗位列表

OpenAI

推理团队工程师

地点:美国 薪资:薪资未公开 日期:2025-02-06

岗位摘要

与机器学习研究员、工程师和产品经理合作,将最新技术投入生产;与研究员合作,通过卓越的工程能力支持前沿研究;引入新技术、工具和架构,以提升模型推理堆栈的性能、延迟、吞吐量和效率;构建工具以识别性能瓶颈和不稳定因素,并设计实施解决方案以解决最高优先级问题

岗位职责

  • 与机器学习研究员、工程师和产品经理合作,将最新技术投入生产
  • 与研究员合作,通过卓越的工程能力支持前沿研究
  • 引入新技术、工具和架构,以提升模型推理堆栈的性能、延迟、吞吐量和效率
  • 构建工具以识别性能瓶颈和不稳定因素,并设计实施解决方案以解决最高优先级问题
  • 优化代码和Azure虚拟机集群,以充分利用硬件的每单位浮点运算能力和GPU内存

任职要求

  • 理解现代机器学习架构,并具备优化其性能(特别是推理性能)的直觉
  • 能够端到端地负责问题,并愿意学习任何缺失的知识以完成任务
  • 至少拥有5年专业软件工程经验
  • 熟悉或能快速掌握PyTorch、NVIDIA GPU及其优化软件栈(如NCCL、CUDA),以及高性能计算技术(如InfiniBand、MPI、NVLink等)
  • 具备架构设计、构建、监控和调试生产级分布式系统的经验。有性能关键型分布式系统经验者优先
  • 曾因规模快速增长而多次重建或大规模重构生产系统
  • 具备自我驱动力,乐于找出最重要的问题进行攻关
  • 态度谦逊,乐于帮助同事,并愿意为团队成功付出一切努力