Compiler Engineer – MLIR / PyTorch Infrastructure
New
Palo Alto, CA / Austin, TX / Vancouver, BC / RemoteFull-TimeMiddle
Salary not disclosed
Apply NowOpens the employer's application page
Job Details
- Experience
- 3+ years
- Required Skills
- PythonPyTorchC++
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.
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.
View Full Description & ApplyYou'll be redirected to the employer's site
