Hardware / Software CoDesign Engineer - 3PActive$342K–$555K

The opportunity

OpenAI’s Hardware organization develops silicon and system-level solutions designed for the unique demands of advanced AI workloads. The team is responsible for building the next generation of AI-native silicon while working closely with software and research partners to…

What you'll do

  • Co-design future hardware for programmability and performance with our hardware vendors

  • Assist hardware vendors in developing optimal kernels and add support for it in our compiler

  • Develop performance estimates for critical kernels for different hardware: configurations and drive decisions on compute core and memory hierarchy features

  • Build system performance models at different abstraction levels and carry out: analysis to drive decisions on scale up, scale out, front end networking

  • Work with machine learning engineers, kernel engineers and compiler: developers to understand their vision and needs from high performance accelerators

  • Manage communication and coordination with internal and external partners

What they're looking for

  • Influence the roadmap of hardware partners to optimize them for OpenAI’s workloads.
  • Evaluate potential partners’ accelerators and platforms.
  • As the scope of the role and team grows, understand and influence roadmaps: for hardware partners for our datacenter networks, racks, and buildings.
  • + years of industry experience, including experience harnessing compute at: scale and optimizing ML platform code to run efficiently on target hardware.