Mythic
Compiler Engineer – MLIR / PyTorch Infrastructure
Palo Alto, CA
Sponsorship not specifiedDetected 293 days ago
C++Machine LearningPyTorchLLMsRoboticsSignal Processing
About the role
- Join us in advancing the MLIR ecosystem at Mythic.
- 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.
Responsibilities
- 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.
- Build conversion paths between MLIR and Mythic's custom low-level IR to keep both flows operational during migration.
- Collaborate with a team spanning hardware, runtime, and compiler design.
- We hire smart, capable individuals, provide the tools and support they need, and trust them to deliver.
Requirements
- 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.
Nice to have
- Architect the migration of the existing compiler flow into MLIR, defining dialects, passes, and lowering strategies.
- 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.
Compensation
- Competitive compensation, equity, and comprehensive benefits.
Benefits
- Competitive compensation, equity, and comprehensive benefits.
Company info
- 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.
- At Mythic, we foster a collaborative and respectful environment where people can do their best work.
This listing is sourced directly from Mythic's careers page and normalized into a canonical job model.