AI Compiler Software Engineer

Ampere; Santa Clara (CA); Hybrid; USD 140,000 - 235,000; Full time

Job summary

Ampere is seeking an AI Compiler Engineer to develop advanced compilers for its upcoming products. The role involves significant contributions to the AI compiler toolchain and collaboration with architecture teams to enhance performance. The ideal candidate will have extensive experience in compiler technologies and machine learning frameworks.

Benefits

Medical, dental, vision insurance

401K plan

Unlimited flextime

10+ paid holidays

Healthy snacks and drinks

Qualifications

• 8 years experience with Bachelor's or 6 years with Master's in relevant fields.
• Strong proficiency in C/C++ and Python with at least 6 years of hands-on experience.

Responsibilities

• Develop and deliver functional AI compilers ensuring robust performance.
• Collaborate with architecture team to enhance AI accelerator.

Skills

C/C++

Python

Machine Learning

Deep Learning

Linux

Git

Education

Bachelor's degree in Electrical Engineering

Master’s degree

Tools

MLIR

LLVM

Job description

Recognized by Fast Company’s 2023 100 Best Workplaces for Innovators List, Ampere is a semiconductor design company for a new era, leading the future of computing with an innovative approach to CPU design focused on high-performance, energy-efficient, sustainable cloud computing. By providing a new level of predictable performance, efficiency, and sustainability, Ampere is working with leading cloud suppliers and a growing partner ecosystem to deliver cloud instances, servers, and embedded/edge products that can handle the compute demands of today and tomorrow.

Join us at Ampere and work alongside a passionate and growing team — we’d love to have you apply!

The Role

Ampere is looking for an enthusiastic and highly-skilled AI Compiler Engineer to join our expanding Compiler Team. In this role, you will develop advanced compilers for Ampere's upcoming AmpereOne Aurora products. Your responsibilities will include making substantial contributions to the entire AI compiler toolchain, which may include language extension, IR lowering, code generation, and compiler-related optimizations. You will work closely with Ampere’s architecture and performance teams to develop a “best in class” algorithm-compiler-hardware ecosystem.

What you'll achieve

1. Successfully develop and deliver a functional AI compiler (either ahead-of-time or just-in-time) along with an associated runtime environment, ensuring robust performance and reliability.
2. Drive the Ampere Computing AI compiler software development, showcasing innovation and excellence in Ampere’s compiler technology.
3. Collaborate with the architecture team to enhance the Ampere Aurora AI accelerator, contributing to cutting-edge hardware-software integration.
4. Design and implement advanced solutions and enhancements for prominent machine learning ecosystems such as PyTorch, JAX, vLLM, and ONNX (C++), significantly boosting their performance and capabilities.
5. Conduct in-depth analyses of AI-related compiler optimizations and processes, leading to substantial performance improvements.

About you

• Bachelor's degree in Electrical Engineering, Computer Engineering, Computer Science, or a related technical field & 8 years of related experience; or Master’s degree & 6 years.
• Strong proficiency in C/C++ and Python languages: At least 6 years of hands-on experience.
• Knowledge of compiler technologies: Experience in control flow and data flow analysis, typical compiler optimizations, and IR lowering.
• Familiarity with MLIR dialect and LLVM IR: Minimum 5 years of LLVM-related compiler development experience.
• Understanding of Machine Learning and Deep Learning frameworks such as PyTorch, TensorFlow, Llama.cpp, and vLLM.
• Proficiency with essential tools like Linux and Git.
• Ability to thrive in a startup environment, taking initiative and achieving impactful results.

What we’ll offer

At Ampere, we believe in taking care of our employees and providing a competitive total rewards package, including base pay, bonus, long-term incentives, and comprehensive benefits. The full base pay range for this role is between $140,500 and $234,500, or $152,500 and $254,000 in the San Francisco Bay Area. Our benefits include:

• Medical, dental, vision insurance, income protection, and a 401K plan.
• Unlimited flextime and 10+ paid holidays for work-life balance.
• Healthy snacks, energizing espresso, and drinks to keep you focused.

We foster an inclusive culture that empowers our employees to grow and innovate. We are passionate about industry-leading, sustainable, cloud-native designs and look forward to sharing more about our career opportunities through the interview process.

#LI-DR
#LI-Hybrid

Ampere is an inclusive and equal opportunity employer. All qualified applicants will receive consideration regardless of race, color, national origin, citizenship, religion, age, veteran/military status, sex, sexual orientation, gender identity, or disability.