Compiler Engineer – MLIR, PyTorch Infrastructure

🕒 March 19

🏄 California – Remote

🏄 California – Remote

⏰ Full Time

🟡 Mid-level

🟠 Senior

👷🏻‍♀️ Engineer

🦅 H1B Visa Sponsor

This company has sponsored H1B visas in the past.

Apply Now

Find Similar Remote Jobs

Set Up Email Alert For New Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

Mythic

WebsiteLinkedInAll Job Openings

11 - 50 employees

Founded 2012

🤖 Artificial Intelligence

🔧 Hardware

💰 $70M Series C on 2021-05

Artificial Intelligence • Hardware • Robotics

Mythic is the leading high-performance analog computing company that focuses on delivering cost-efficient and energy-efficient edge AI computing power across various sectors including robotics, defense, and consumer devices. The company's innovative compute-in-memory technology solves the limitations of traditional digital computing by storing AI parameters directly in the processor, significantly enhancing AI inference applications. Mythic aims to empower industries with advancements in performance scalability and efficiency, providing a robust platform for the next generation of AI applications.

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

🎯 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.