OpenAI is looking to develop and optimize low-level systems to drive the performance of AI workloads on cutting-edge hardware, ensuring safety, ethical considerations, and inclusivity in its mission to create powerful AI tools that address global challenges.
Requirements
- Proficiency in systems programming languages such as Rust and C++
- Experience with scripting languages like Python
- Strong understanding of compiler development, kernel design, or accelerator programming
- Experience in runtime systems, distributed systems, or high-performance simulation
- Familiarity with hardware architecture and optimization techniques
- Ability to analyze and optimize system bottlenecks related to I/O, memory hierarchy, and interconnects
- Experience working in fast-paced, collaborative environments with evolving hardware and ML systems
Responsibilities
- Design and implement APIs and runtime components to manage computation and data movement across heterogeneous ML workloads
- Contribute to compiler infrastructure, including developing optimizations and compiler passes to support new hardware features
- Engineer and optimize compute and data kernels for correctness, high performance, and portability in simulation and production environments
- Profile system performance, identify bottlenecks, and implement improvements at local and distributed scales
- Develop simulation infrastructure to validate runtime behavior, test hardware and system changes, and support early hardware development phases
- Collaborate with hardware and research teams to rapidly deploy runtime and compiler updates across supercomputing clusters
- Work across a diverse technology stack, primarily using Rust and Python, influencing architecture decisions within the training framework
Other
- Excellent communication and teamwork abilities
- Strong problem-solving skills and a deep curiosity about large-scale systems
- 3 days per week in the office in San Francisco, CA
- Relocation assistance available
- Equal opportunity employer, celebrating diversity and committed to creating an inclusive environment for all employees
