Taro Logo

Graph Compiler Engineer

AI research and deployment company dedicated to ensuring that general-purpose artificial intelligence benefits all of humanity.
$245,000 - $385,000
Backend
Senior Software Engineer
Hybrid
5+ years of experience
This job posting may no longer be active. You may be interested in these related jobs instead:

Description For Graph Compiler Engineer

OpenAI is seeking a Graph Compiler Engineer to join their Platform ML team. This role focuses on building the ML side of OpenAI's state-of-the-art internal training framework used to train cutting-edge models. The successful candidate will work on distributed model execution, interfaces, and implementation for model code, training, and inference.

Key Responsibilities:

  • Work on OpenAI's internal training framework to enable easy graph capture of research experiments
  • Apply advanced optimizations to the graph to maximize hardware resource utilization
  • Collaborate with researchers at the cutting edge of ML research
  • Improve training throughput and researcher productivity

Ideal Candidate:

  • Strong software engineering skills with proficiency in Python
  • Experience running machine learning experiments
  • Experience working on a machine learning compiler
  • Deep knowledge of GPU performance and systems-level optimization

The role offers an exciting opportunity to work on OpenAI's training framework and contribute to accelerating progress towards AGI. Small improvements in hardware utilization for large training runs can produce significant cost savings, while enabling researchers to iterate quickly on their ideas will lead to substantial improvements in model capabilities and compute efficiency.

OpenAI offers a competitive compensation package, including a salary range of $245K – $385K, generous equity, and comprehensive benefits. The company is committed to diversity, equality, and creating an inclusive work environment.

This is a hybrid position based in San Francisco, CA, requiring 3 days in the office per week. Relocation assistance is available for new employees.

Last updated a year ago

Responsibilities For Graph Compiler Engineer

  • Work on internal training framework to enable easy graph capture of research experiments
  • Apply advanced optimizations to the graph to maximally utilize hardware resources
  • Collaborate with researchers at the cutting edge of ML research
  • Improve training throughput and researcher productivity

Requirements For Graph Compiler Engineer

Python
  • Strong software engineering skills and proficiency in Python
  • Experience running machine learning experiments
  • Experience working on a machine learning compiler
  • Deep knowledge of GPU performance and systems-level optimization

Benefits For Graph Compiler Engineer

Medical Insurance
Dental Insurance
Vision Insurance
401k
Mental Health Assistance
Parental Leave
Education Budget
  • 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 13 company holidays per year
  • Paid parental leave (20 weeks) and family-planning support
  • Annual learning & development stipend ($1,500 per year)
  • Generous equity

Interested in this job?