ML Compiler Engineer

IC Resources; California; Remote; USD 120,000 - 170,000; Full time

Job summary

A leading organization is seeking a skilled MLIR/XLA Compiler Engineer to enhance AI workload performance. This role involves optimizing compiler transformations and collaborating with teams to push the boundaries of AI infrastructure, offering a flexible work environment and competitive salary.

Qualifications

• Strong experience in compiler development with MLIR/XLA.
• Proficiency in C++ and Python.
• Understanding of machine learning frameworks like TensorFlow or PyTorch.

Responsibilities

• Develop and optimize compiler transformations.
• Contribute to performance tuning and benchmarking.
• Collaborate with cross-functional teams on compiler improvements.

Skills

Compiler Development

C++

Python

MLIR

XLA

Problem-Solving

Job description

Direct message the job poster from IC Resources

Team Principal at IC Resources - High Level Software / Compilers

MLIR/XLA Compiler Engineer

Location: Remote or Bay Area

Job Type: Full-time

About the Company:

A leading organization at the forefront of AI infrastructure is pushing the boundaries of compiler technology to optimize 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 optimizations, 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 optimize 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 optimizing AI workloads for GPUs, TPUs, or other accelerators.

- Strong problem-solving skills and the ability to work in a fast-paced environment.

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

Seniority level

• Seniority level

  Mid-Senior level

Employment type

• Employment type

  Full-time

Job function

• Job function

  Information Technology and Engineering

• Industries

  Software Development and Semiconductor Manufacturing

Referrals increase your chances of interviewing at IC Resources by 2x

Get notified about new Software Engineer jobs in Santa Clara County, CA.

Software Engineer, AI Platform - New Grad

Mountain View, CA $125,400.00-$188,100.00 2 weeks ago

(General Hire) Software Engineer Graduate (Advertisement Team) - 2025 Start (BS/MS)

San Jose, CA $113,500.00-$250,000.00 2 weeks ago

Full Stack Software Engineer (L4), Product Localization Engineering

Full Stack Software Engineer - Post-training

New Grads 2025 - Software Engineer, Algorithm

San Jose, CA $120,000.00-$165,000.00 8 months ago

Software Engineer 4 - TV & Web Player Platform

Palo Alto, CA $153,000.00-$211,000.00 2 weeks ago

Software Engineer I (Co-Op) United States

San Jose, CA $44,000.00-$130,000.00 1 week ago

New Grads 2025 - General Software Engineer

San Jose, CA $120,000.00-$165,000.00 4 months ago

Software Engineer (L5) - Open Connect Platform

United States $100,000.00-$720,000.00 2 days ago

New College Grad Software Engineer, Software Engineering Development (Apps)

San Jose, CA $92,735.00-$131,300.00 1 week ago

Software Engineer - Intern (Summer 2025)

San Jose, CA $3,000.00-$4,000.00 8 months ago

Associate Software Engineer, Backend Python

Santa Clara, CA $150,000.00-$175,000.00 7 months ago

San Jose, CA $113,400.00-$206,300.00 1 week ago

eCommerce Full Stack Developer (React / Shopify) - On Site

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.