companyOpenAI logo

Hardware and Software Co-Design Engineer

OpenAISan Francisco
Hybrid Full-time

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Experience Level

Entry Level

Qualifications

Qualifications:Ideal candidates should possess a solid foundation in hardware and software co-design principles, along with experience in machine learning frameworks and high-performance computing. A Bachelor's degree in Computer Engineering, Electrical Engineering, or a related field is preferred. Familiarity with compiler optimizations and performance modeling is advantageous, as well as strong analytical skills to assess hardware performance.

About the job

About Our Team

At OpenAI, our Hardware organization is at the forefront of developing cutting-edge silicon and system-level solutions tailored for the specific demands of advanced AI workloads. Our team is dedicated to creating the next generation of AI-native silicon, collaborating closely with software and research partners to co-design hardware that is seamlessly integrated with AI models. We not only deliver production-grade silicon for OpenAI’s supercomputing infrastructure but also innovate custom design tools and methodologies that drive acceleration and optimization specific to AI.

About This Role

As a member of our hardware optimization and co-design team, you will play a crucial role in co-designing future hardware from various vendors, focusing on programmability and high performance. You will partner with our kernel, compiler, and machine learning engineers to comprehend their distinct requirements concerning ML techniques, algorithms, numerical approximations, programming expressivity, and compiler optimizations. Your advocacy for these constraints will help shape and influence future hardware architectures aimed at efficient training and inference for our models. If you are passionate about efficiently distributing large language models across devices, optimizing system-wide networking bottlenecks, and customizing the compute pipeline and memory hierarchy of hardware platforms while simulating workloads at various abstraction levels, then this opportunity is perfect for you!

This position is based in San Francisco, CA, utilizing a hybrid work model of three days in the office each week, with relocation assistance available for new hires.

Key Responsibilities:

  • Collaborate on the co-design of future hardware focusing on programmability and performance with hardware vendors.
  • Support hardware vendors in developing optimal kernels and integrating support within our compiler.
  • Generate performance estimates for critical kernels across diverse hardware configurations, influencing decisions regarding compute core and memory hierarchy features.
  • Create system performance models at various abstraction levels and conduct analyses to guide decisions on scaling and front-end networking.
  • Engage with machine learning engineers, kernel engineers, and compiler developers to align on high-performance accelerator needs.
  • Facilitate communication and coordination with internal and external partners.
  • Shape the roadmap for hardware partners to optimize their products for our AI capabilities.

About OpenAI

OpenAI is a pioneering organization in the field of artificial intelligence, dedicated to ensuring that general AI benefits all of humanity. Our innovative approach combines extensive research with advanced technology development, facilitating breakthroughs in AI applications and hardware that push the boundaries of what is possible.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.