Compiler Engineer – MLIR / PyTorch Infrastructure

Mythic Ai.Com; Palo Alto (CA); On-site; USD 150,000 - 210,000; Full time

Benefits offered by this job

Equity

Competitive salary

Comprehensive benefits

Job summary

Mythic is seeking an experienced MLIR compiler engineer in California to migrate the compiler flow into MLIR, design dialects, and create lowering paths to Mythic’s IR. You will work with hardware and ML teams to enable MLIR integration across analog accelerators and digital subsystems.

Proficiency in C++ and Python is required, along with hands-on MLIR experience and PyTorch compiler knowledge. The role offers competitive compensation, equity, and comprehensive benefits.

Qualifications

• 3+ years of experience in compiler or high-performance systems development.
• Proficiency in modern C++ (C++14/17/20) and Python.
• Direct, hands-on experience with MLIR, including dialect design, compiler passes, or lowering pipelines.
• Strong understanding of compiler IRs and transformations.
• Experience with PyTorch compiler technologies, Torch-MLIR and integration paths with PyTorch 2.0 (TorchDynamo, TorchInductor).

Responsibilities

• Architect the migration of the existing compiler flow into MLIR, defining dialects, passes, and lowering strategies.
• Build conversion paths between MLIR and Mythic’s custom low-level IR to keep both flows operational during migration.
• Define validation infrastructure within MLIR, including interpretation or execution paths for simulation and debugging.
• Enable compilation by extending MLIR integration across analog accelerators and digital subsystems.
• Leverage Torch-MLIR where PyTorch inputs are available, and guide future integration with PyTorch 2.0 compiler technologies (TorchInductor, TorchDynamo, Torch-MLIR)
• Shape the MLIR and PyTorch integration strategy for novel AI hardware.
• Collaborate with a team spanning hardware, runtime, and compiler design.

Skills

C++

Python

Compiler development

MLIR knowledge

PyTorch integration

Performance optimization

Dialect design

Team collaboration

Tools

MLIR

Torch-MLIR

TorchDynamo

Job description

About us

Mythic is building the future of AI computing with breakthrough analog technology that delivers 100× the performance of traditional digital systems at the same power and cost. This unlocks bigger, more capable models and faster, more responsive applications—whether in edge devices like drones, robotics, and sensors, or in cloud and data center environments. Our technology powers everything from large language models and CNNs to advanced signal processing, and is engineered to operate from –40 °C to +125 °C, making it ideal for industrial, automotive, aerospace, and defense.

We’ve raised over $100M from world-class investors including Softbank, Threshold Ventures, Lux Capital, and DCVC, and secured multi-million-dollar customer contracts across multiple markets.

About the role

Join us in advancing the MLIR ecosystem at Mythic. You’ll help extend our existing high-level dialects and design a new hardware-aware low-level dialect, building conversion paths that bridge to our current IRs. Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an MLIR-based stack that unifies hardware constraints with developer needs and accelerates adoption of modern ecosystems.

• Architect the migration of the existing compiler flow into MLIR, defining dialects, passes, and lowering strategies.
• Build conversion paths between MLIR and Mythic’s custom low-level IR to keep both flows operational during migration.
• Define validation infrastructure within MLIR, including interpretation or execution paths for simulation and debugging.
• Enable compilation by extending MLIR integration across analog accelerators and digital subsystems.
• Leverage Torch-MLIR where PyTorch inputs are available, and guide future integration with PyTorch 2.0 compiler technologies (TorchInductor, TorchDynamo, Torch-MLIR)
• 3+ years of experience in compiler or high-performance systems development.
• Proficiency in modern C++ (C++14/17/20) and Python.
• Direct, hands-on experience with MLIR, including dialect design, compiler passes, or lowering pipelines.
• Strong understanding of compiler IRs and transformations, with the ability to reason about lowering from high-level ops to hardware-aware representations.
• Experience architecting complete MLIR flows: from frontend dialects down to hardware‑aware dialects, including conversion to and from existing IRs.
• Familiarity with PyTorch compiler technologies, especially Torch-MLIR and integration paths with PyTorch 2.0 (TorchDynamo, TorchInductor).
• Knowledge of dataflow architectures, scheduling, and memory orchestration.
• Background in heterogeneous or specialized accelerators (e.g., analog compute, NPUs, GPUs, DSPs).
• Shape the MLIR and PyTorch integration strategy for novel AI hardware.
• Opportunity to contribute upstream and influence the broader ML compiler ecosystem.
• Collaborate with a team spanning hardware, runtime, and compiler design.
• Competitive compensation, equity, and comprehensive benefits.

At Mythic, we foster a collaborative and respectful environment where people can do their best work. We hire smart, capable individuals, provide the tools and support they need, and trust them to deliver. Our team brings a wide range of experiences and perspectives, which we see as a strength in solving hard problems together. We value professionalism, creativity, and integrity, and strive to make Mythic a place where every employee feels they belong and can contribute meaningfully.