ML Compiler Engineer

MLIR/XLA Compiler Engineer

Location: Remote or Bay Area

Job Type: Full-time

About the Company:
A leading organisation at the forefront of AI infrastructure is pushing the boundaries of compiler technology to optimise performance and efficiency for machine learning workloads. The team is dedicated to building cutting-edge solutions that bridge the gap between AI models and high-performance hardware.

The Role:
The company is seeking an experienced MLIR/XLA Compiler Engineer to join its growing team. This role involves contributing to the design and implementation of compiler optimisation's, transformations, and code generation techniques to enhance AI/ML performance. The engineer will collaborate closely with hardware engineers, ML researchers, and software developers to drive innovation in AI acceleration.

Key Responsibilities:

• Develop and optimise compiler transformations using MLIR and XLA to enhance AI workload performance.  
• Work on lowering AI/ML models to hardware-efficient representations.  
• Contribute to performance tuning and benchmarking for various target architectures.  
• Collaborate with cross-functional teams to ensure seamless integration of compiler improvements.  
•  Stay up-to-date with the latest advancements in compiler and AI optimization techniques.

What the Company is Looking For 

• Strong experience in compiler development, particularly using MLIR, XLA, LLVM, or similar technologies.  
• Proficiency in C++ and Python for compiler and toolchain development.  
• Understanding of machine learning frameworks such as TensorFlow, PyTorch, or JAX.  
• Experience optimising AI workloads for GPUs, TPUs, or other accelerators.  
• Strong problem-solving skills and the ability to work in a fast-paced environment.  

Why Jin the Company?

• Work on groundbreaking AI compiler technology with a world-class team.  
• Opportunity to shape the future of AI acceleration.  
• Competitive salary and benefits package.  
• Flexible work environment with remote opportunities.

If you’re a ML Compiler Engineer looking for an exciting new challenge within a great company, then please apply today to learn more!

To find out more about this and other Software opportunities across the UK/US/EU, please contact Mitch Wheaton at IC Resources.