AI Hardware & Software Co-Design Engineer

OpenAI; San Francisco (CA); Hybrid; USD 342,000 - 555,000; Full time

Job summary

OpenAI is looking for an Engineer to join their hardware optimization and co-design team in San Francisco. In this role, you will co-design hardware for programmability and performance by collaborating with hardware vendors and engineers. You'll evaluate potential partners' accelerators and influence hardware architectures toward optimal solutions for AI models.

The position offers a hybrid work model and competitive compensation in the range of $342K to $555K.

Qualifications

• Experience optimizing ML platform code on target hardware.
• Deep understanding of GPU and AI accelerators.
• Strong experience in software/hardware co-design.

Responsibilities

• Co-design future hardware for programmability and performance.
• Assist vendors in developing optimal kernels.
• Develop performance estimates for different hardware configurations.

Skills

Software/hardware co-design

GPU and/or AI accelerators expertise

Coding in C/C++ and Python

Machine Learning accuracy with low-precision formats

Education

4+ years industry experience

PhD in Computer Science or Engineering

Tools

CUDA

Triton

Job description

OpenAI is looking for an Engineer to join their hardware optimization and co-design team in San Francisco. In this role, you will co-design hardware for programmability and performance by collaborating with hardware vendors and engineers. You'll evaluate potential partners' accelerators and influence hardware architectures toward optimal solutions for AI models.

The position offers a hybrid work model and competitive compensation in the range of $342K to $555K.