Logo
Ampere

AI Compiler Software Engineer

Ampere, Santa Clara

Save Job

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 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, showcasing innovation and excellence in Ampere’s compiler technology. Collaborate 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 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. #J-18808-Ljbffr