Base Salary
$380k - $555k/yr
Responsibilities
- Design and implement efficient collective operations in C++ and CUDA in collaboration with ML researchers.
- Ensure large training jobs fully utilize the network transports available in the supercomputers.
- Develop simulations to inform future supercomputer network designs.
Requirements
- Background in low-level, performance-critical software.
- Experience writing distributed algorithms using RDMA.
- Comfort writing low-level, performance-sensitive CPU and/or GPU code.
- Familiarity with network simulation techniques.
- Experience with collective communication is a bonus.
Benefits
- Hybrid work model with 3 days in the San Francisco office per week
- Relocation assistance for new employees
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.