AI Hardware Co-Design Engineer — Hybrid (SF)

OpenAI; United States; Hybrid; USD 150,000 - 190,000; Full time

Benefits offered by this job

Relocation assistance

Job summary

OpenAI is hiring an Engineer for the hardware optimization and co-design team in San Francisco, CA. You will collaborate with kernel, compiler, and ML engineers to tailor hardware for AI workloads, focusing on ML techniques, numerical approximations, and compiler optimizations.

The role involves working with vendors to shape future hardware architectures for efficient training and inference. The position offers a hybrid work model (3 days in the office) and relocation assistance for new

Responsibilities

• Co-design future hardware for programmability and performance with our 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 and drive decisions on compute core and memory hierarchy

Job description

OpenAI is hiring an Engineer for the hardware optimization and co-design team in San Francisco, CA. You will collaborate with kernel, compiler, and ML engineers to tailor hardware for AI workloads, focusing on ML techniques, numerical approximations, and compiler optimizations.

The role involves working with vendors to shape future hardware architectures for efficient training and inference. The position offers a hybrid work model (3 days in the office) and relocation assistance for new