返回岗位列表

OpenAI

加速器团队软件工程师

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

岗位摘要

在新型探索性加速器平台上原型化并启用OpenAI的AI软件栈;针对多样化硬件环境优化大规模模型性能(如LLM、推荐系统、分布式AI工作负载);为新兴加速器开发定制化的内核、分片机制和系统扩展策略

岗位职责

  • 在新型探索性加速器平台上原型化并启用OpenAI的AI软件栈
  • 针对多样化硬件环境优化大规模模型性能(如LLM、推荐系统、分布式AI工作负载)
  • 为新兴加速器开发定制化的内核、分片机制和系统扩展策略
  • 在模型代码层(如PyTorch)及以下进行协作优化,以提升在非传统硬件上的性能
  • 执行系统级性能建模、调试瓶颈并推动端到端优化
  • 与硬件团队和供应商合作,评估现有平台的替代方案,并使软件栈适应其架构
  • 为前沿AI工作负载的运行时改进、计算/通信重叠和扩展工作做出贡献

任职要求

  • 拥有3年以上AI基础设施工作经验,涉及内核、系统或软硬件协同设计
  • 具备数据中心规模AI加速器平台(如TPU、定制芯片、探索性架构)的实践经验
  • 深刻理解内核、分片、运行时系统或分布式扩展技术
  • 熟悉为硬件效率优化LLM、CNN或推荐模型
  • 具备性能建模、系统调试和为新型架构适配软件栈的经验
  • 接触过移动加速器者优先,但更青睐具备数据中心规模AI硬件启用经验者
  • 能够跨多个技术栈层级工作,快速原型化解决方案,并在新兴硬件领域的模糊性中导航