lever
Compiler Engineer – MLIR / PyTorch Infrastructure
Mythic
Remote, USA
Full-time
Mid · 3+ years experience
Remote
Discovered 1 weeks ago
MLIRPyTorchC++Python
Free
Job Fit Check
Base Career helps you apply smarter for this job.
?%
Ready to ScanKey skills for this role
MLIRPyTorchC++
Key Skills for This Role
MLIRPyTorchC++Python
Full Job Posting
- 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.
Apply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
More jobs at Mythic
Digital Media Buying Analyst
Charlotte, USA
Discovered YesterdayFull-time
Design Verification Engineer
Austin, USA
Discovered 1 weeks agoFull-time
RTL Design Engineer
Austin, USA
Discovered 1 weeks agoFull-time
Compiler Engineer - Algorithmic Workloads Compilation
, USA
Discovered 1 weeks agoFull-time
Senior Silicon Emulation Engineer
Austin, USA
Discovered 1 weeks agoFull-time
Social Creative Lead
, USA
Discovered 1 weeks agoFull-time
Senior Comms Strategist
Charlotte, USA
Discovered 1 weeks agoFull-time
Account Director
Charlotte, USA
Discovered 1 weeks agoFull-time