Mythic

Pioneering Analog Compute for AI

Compiler Engineer – MLIR, PyTorch Infrastructure

EngineerEngineerFull TimeRemoteSeniorTeam 11-50Since 2012H1B SponsorCompany Site LinkedIn

Location

California

Posted

1 day ago

Salary

Not specified

Seniority

Senior

Bachelor Degree3 yrs expEnglishC++PythonPyTorch

Job Description

• 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)

Job Requirements

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.

Benefits

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.