Mythic AI Logo

Mythic AI

Compiler Engineer – MLIR / PyTorch Infrastructure

Reposted 24 Days Ago
In-Office or Remote
Hiring Remotely in Austin, TX, USA
Mid level
In-Office or Remote
Hiring Remotely in Austin, TX, USA
Mid level
Architect and build MLIR ecosystems, extending dialects while ensuring conversion paths. Collaborate with hardware engineers to integrate with PyTorch and modern frameworks.
The summary above was generated by AI
About us
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.

About the role
Join us in advancing the MLIR ecosystem at Mythic. 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. 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.

Here's what you will do

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

Here's the background we hope you will have

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

The following would be nice to have, but is not required

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

What we offer

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

At Mythic, we foster a collaborative and respectful environment where people can do their best work. We hire smart, capable individuals, provide the tools and support they need, and trust them to deliver. Our team brings a wide range of experiences and perspectives, which we see as a strength in solving hard problems together. We value professionalism, creativity, and integrity, and strive to make Mythic a place where every employee feels they belong and can contribute meaningfully.

HQ

Mythic AI Austin, Texas, USA Office

Austin, TX, United States

Similar Jobs

50 Minutes Ago
Remote or Hybrid
USA
41K-69K Annually
Mid level
41K-69K Annually
Mid level
Machine Learning • Payments • Security • Software • Financial Services
Lead escalation point for enterprise service desk: provide multi-channel (chat/ticket/phone) technical support, troubleshoot hardware/software/network issues, manage incidents in ServiceNow, coach junior analysts, identify process improvements, and uphold security and customer service standards in a remote environment.
Top Skills: Chat Support PlatformsItsm PlatformsNetworking FundamentalsRemote Desktop ToolsServicenowTicketing Systems
50 Minutes Ago
Remote or Hybrid
USA
91K-186K Annually
Senior level
91K-186K Annually
Senior level
Machine Learning • Payments • Security • Software • Financial Services
As a Quality Assurance Architect in Test Data Management, design and implement test data solutions, establish standards, and ensure compliance across non-production environments.
Top Skills: BroadcomDelphixInformatica Tdm
2 Hours Ago
Remote or Hybrid
United States
60K-81K Annually
Junior
60K-81K Annually
Junior
Cloud • Fintech • Software • Business Intelligence • Consulting • Financial Services
Support financial advisory engagements for Federally Qualified Health Centers by analyzing financial and operational data, assessing finance performance, recommending process improvements, assisting with budgeting/forecasting, configuring financial/EHR systems, and collaborating with clients and senior consultants to improve revenue cycle, cash flow, and reporting.
Top Skills: AthenahealthEclinicalworks (Ecw)EpicMicrosoft Dynamics Great Plains (Great Plains)ExcelMicrosoft OutlookMicrosoft TeamsMicrosoft WordMipNextgenQuickbooksSage Intacct

What you need to know about the Austin Tech Scene

Austin has a diverse and thriving tech ecosystem thanks to home-grown companies like Dell and major campuses for IBM, AMD and Apple. The state’s flagship university, the University of Texas at Austin, is known for its engineering school, and the city is known for its annual South by Southwest tech and media conference. Austin’s tech scene spans many verticals, but it’s particularly known for hardware, including semiconductors, as well as AI, biotechnology and cloud computing. And its food and music scene, low taxes and favorable climate has made the city a destination for tech workers from across the country.

Key Facts About Austin Tech

  • Number of Tech Workers: 180,500; 13.7% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Dell, IBM, AMD, Apple, Alphabet
  • Key Industries: Artificial intelligence, hardware, cloud computing, software, healthtech
  • Funding Landscape: $4.5 billion in VC funding in 2024 (Pitchbook)
  • Notable Investors: Live Oak Ventures, Austin Ventures, Hinge Capital, Gigafund, KdT Ventures, Next Coast Ventures, Silverton Partners
  • Research Centers and Universities: University of Texas, Southwestern University, Texas State University, Center for Complex Quantum Systems, Oden Institute for Computational Engineering and Sciences, Texas Advanced Computing Center

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account