AI Compiler Software Engineer

Description

Invent the future with us.  

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 be making substantial contributions to entire AI compiler toolchain, which may include language extension, IR lowering, code generation and compiler related optimizations. You will have a chance to work closely with Ampere’s architecture and performance teams to develop a “best in class” algorithm-compiler-hardware ecosystem.

What You'll Achieve

• 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.
• Drive the Ampere Computing AI compiler software development, showcase innovation and excellence in Ampere’s compiler technology.
• Work hand-in-hand with the architecture team to enhance the Ampere Aurora AI accelerator, contributing to cutting-edge hardware-software integration.
• 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.
• Conduct in-depth analyses of AI related compiler optimizations and process, leading to substantial performance improvements.
  
  

About You

• Bachelors 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: Bring your expertise and passion for programming with at least 6years of hands-on experience in C/C and Python.
• Knowledge of compiler technologies: Strong experience in control flow and data flow analysis, and typical compiler optimizations and IR lowering.
• Familiarity of MLIR dialect and LLVM IR: Minimum 5 years of LLVM related compiler development experience is required in this position.
• Understand Machine Learning and Deep Learning: Hands-on experience in popular frameworks such as PyTorch, TensorFlow, Llama.cpp, and vLLM.
• Knowledge of Essential Tools: Demonstrate your command of vital development tools such as Linux, and Git, ensuring smooth, efficient workflows and high-quality results.
• Adaptability in a Startup Environment: Thrive in our fast-paced, dynamic startup culture where your ability to take initiative and achieve impactful results is highly valued and rewarded.
  
  

What we’ll offer: 

At Ampere we believe in taking care of our employees and providing a competitive total rewards package that includes base pay, bonus (i.e., variable pay tied to internal company goals), long-term incentive, and comprehensive benefits. The full base pay range for this role is between $140,500 and $234,500, except in the San Francisco Bay Area where the range is between $152,500 and $254,000.

Our benefits include health, wellness, and financial programs that support employees through every stage of life, with full benefits eligibility at 20 hours per week.

Benefit Highlights Include

• Premium medical insurance, dental insurance, vision insurance, as well as income protection and a 401K retirement plan, so that you can feel secure in your health and financial future.
• Unlimited Flextime and 10 paid holidays so that you can embrace a healthy work-life balance.
• A variety of healthy snacks, energizing espresso, and refreshing drinks to keep you fueled and focused throughout the day.
  
  

And there is much more than compensation and benefits. At Ampere, we foster an inclusive culture that empowers our employees to do more and grow more. We are passionate about inventing industry leading cloud-native designs that contribute to a more sustainable future. We are excited to share more about our career opportunities with you through the interview process.

Ampere is an inclusive and equal opportunity employer and welcomes applicants from all backgrounds. All qualified applicants will receive consideration for employment without regard to race, color, national origin, citizenship, religion, age, veteran and/or military status, sex, sexual orientation, gender, gender identity, gender expression, physical or mental disability, or any other basis protected by federal, state or local law.