HW/SW Co-Design Engineer
Description For HW/SW Co-Design Engineer
OpenAI is seeking a Hardware/Software Co-Design Engineer to join their Research Platform team in San Francisco. This role is crucial in optimizing hardware for AI workloads and collaborating with partners to develop next-generation deep learning accelerators. The position offers a competitive salary range of $360K-$530K plus equity and comprehensive benefits. The ideal candidate will have 4+ years of experience in hardware optimization, deep understanding of GPU architectures, and expertise in software/hardware co-design. This is a hybrid role requiring 3 days in office, working on cutting-edge AI infrastructure that powers OpenAI's mission to develop safe and beneficial AGI. The role combines technical depth in hardware architecture with strategic partnership management, offering a unique opportunity to influence the future of AI computing infrastructure. The position includes extensive benefits such as comprehensive healthcare, unlimited PTO, parental leave, and educational stipends, making it an attractive opportunity for experienced engineers passionate about advancing AI technology.
Responsibilities For HW/SW Co-Design Engineer
- Co-design future hardware for programmability and performance with 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
- Work with machine learning engineers, kernel engineers and compiler developers
- Manage communication and coordination with internal and external partners
- Influence the roadmap of hardware partners to optimize them for OpenAI's workloads
- Evaluate potential partners' accelerators and platforms
- Build simulations and performance models
- Influence roadmaps for hardware partners for datacenter networks, racks, and buildings
Requirements For HW/SW Co-Design Engineer
- 4+ years of industry experience in harnessing compute at scale or building semiconductors
- Strong experience in software/hardware co-design
- Deep understanding of GPU and/or other AI accelerators
- Experience with CUDA or related accelerator programming language
- Experience driving Machine Learning accuracy with low precision formats
- Experience aligning future hardware with HPC infrastructure
- Familiar with fundamentals of deep learning computing and chip microarchitecture
- Able to actively collaborate with ML engineers, kernel writers and compiler developers
Benefits For HW/SW Co-Design Engineer
- Medical, dental, and vision insurance for you and your family
- Mental health and wellness support
- 401(k) plan with 50% matching
- Unlimited time off and 18+ company holidays per year
- Paid parental leave (24 weeks birth-parent, 20 weeks parental)
- Annual learning & development stipend ($1,500 per year)
- Equity compensation
- Relocation assistance
Jobs Related To OpenAI HW/SW Co-Design Engineer
