Base Salary
$266k - $455k/yr
Responsibilities
- Design and build APIs and runtime components for computation and data movement across heterogeneous ML workloads.
- Develop compiler optimizations and compiler passes for evolving hardware.
- Engineer and optimize compute and data kernels for correctness, performance, and portability.
- Profile and optimize bottlenecks involving I/O, memory hierarchy, and interconnects at local and distributed scales.
- Develop simulation infrastructure to validate runtime behavior and training-stack changes and support early hardware and system development.
- Deploy runtime and compiler updates to new supercomputing builds in collaboration with hardware and research teams.
Requirements
- Proficiency in systems programming such as Rust or C++ and scripting languages such as Python.
- Experience in one or more of compiler development, kernel authoring, accelerator programming, runtime systems, distributed systems, or high-performance simulation.
- Curiosity about large-scale systems and optimizing them for speed, simplicity, and reliability.
- Interest in collaborative work across evolving hardware and ML system requirements.
- Values engineering excellence, technical leadership, and thoughtful system design.
Benefits
- Hybrid work model with 3 days per week in the San Francisco office.
- Relocation assistance is available.
Categories
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.