AI Compiler Software Engineer

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. 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.

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.