Base Salary
$295k - $380k/yr
Responsibilities
- Prototype and enable OpenAI’s AI software stack on new and exploratory accelerator platforms.
- Optimize large-scale model performance and distributed AI workloads across diverse hardware environments.
- Develop kernels, sharding mechanisms, and system scaling strategies for emerging accelerators.
- Collaborate on model-level and low-level optimizations, including PyTorch-based workloads.
- Model system performance, debug bottlenecks, and drive end-to-end optimization.
- Work with hardware teams and vendors to evaluate alternative platforms and adapt the software stack to their architectures.
- Contribute to runtime improvements, compute and communication overlapping, and scaling for frontier AI workloads.
Requirements
- At least 3 years of experience in AI infrastructure, including kernels, systems, or hardware-software co-design.
- Hands-on experience with data-center-scale AI accelerator platforms such as TPUs, custom silicon, or exploratory architectures.
- Strong understanding of kernels, sharding, runtime systems, or distributed scaling techniques.
- Experience optimizing LLMs, CNNs, or recommender models for hardware efficiency.
- Experience with performance modeling, system debugging, and adapting software stacks to novel architectures.
- Exposure to mobile accelerators is welcome, while data-center-scale AI hardware experience is preferred.
- Ability to work across multiple levels of the stack, rapidly prototype solutions, and navigate ambiguity during early hardware bring-up.
About OpenAI
OpenAI is an AI research and deployment company dedicated to ensuring that general-purpose artificial intelligence benefits all of humanity. AI is an extremely powerful tool that must be created with safety and human needs at its core. OpenAI is dedicated to putting that alignment of interests first — ahead of profit. To achieve our mission, we must encompass and value the many different perspectives, voices, and experiences that form the full spectrum of humanity. Our investment in diversity, equity, and inclusion is ongoing, executed through a wide range of initiatives, and championed and supported by leadership. At OpenAI, we believe artificial intelligence has the potential to help people solve immense global challenges, and we want the upside of AI to be widely shared. Join us in shaping the future of technology.