RemoteLeaf Logo
Log in Sign up

Compiler Engineer – MLIR / PyTorch Infrastructure

5 months ago
United States, Canada
Full-time
Mid Level
AI Engineer
Software Development
C++ Python PyTorch
Apply Now
Mythic

Mythic

Mythic revolutionizes edge AI with power-efficient analog compute in memory technology, enabling affordable and powerful AI solutions from data centers to edge devices.

Semiconductors & Semiconductor Equipment
51-250
Founded 2012
$165M raised
View All Jobs 3

Description

  • Architect the migration of the existing compiler flow into MLIR by defining dialects, passes, and lowering strategies.
  • Build conversion paths between MLIR and Mythic’s custom low-level IR to keep both compilation flows operational during migration.
  • Define validation infrastructure within MLIR, including interpretation or execution paths for simulation and debugging.
  • Enable compilation across analog accelerators and digital subsystems through MLIR integration.
  • Leverage Torch-MLIR for PyTorch inputs when available.
  • Guide future integration with PyTorch 2.0 compiler technologies, including TorchInductor and TorchDynamo.
  • Collaborate closely with hardware engineers and ML developers on compiler and framework integration.
  • Help shape the MLIR ecosystem and long-term compiler architecture at Mythic.

Requirements

  • 3+ years of experience in compiler or high-performance systems development.
  • Proficiency in modern C++ (C++14/17/20) and Python.
  • Hands-on experience with MLIR, including dialect design, compiler passes, or lowering pipelines.
  • Strong understanding of compiler IRs and transformations.
  • Ability to reason about lowering from high-level operations to hardware-aware representations.
  • Experience architecting complete MLIR flows from frontend dialects to hardware-aware dialects, including conversion to and from existing IRs (nice to have).
  • Familiarity with PyTorch compiler technologies, especially Torch-MLIR and integration paths with PyTorch 2.0, TorchDynamo, and TorchInductor (nice to have).
  • Knowledge of dataflow architectures, scheduling, and memory orchestration (nice to have).
  • Background in heterogeneous or specialized accelerators such as analog compute, NPUs, GPUs, or DSPs (nice to have).

Benefits

  • Competitive compensation and equity.
  • Comprehensive benefits.
  • Opportunity to shape the MLIR and PyTorch integration strategy for novel AI hardware.
  • Chance to contribute upstream and influence the broader ML compiler ecosystem.
  • Collaborative team spanning hardware, runtime, and compiler design.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Caseware

Senior AI Software Engineer

Caseware 251-1K Internet Software & Services

Caseware is hiring a fully remote AI Platform Architect in Colombia to define, build, and operate scalable, secure, and trustworthy AI capabilities for its global accounting and audit software platform.

Colombia Full-time Lead AI Engineer
AWS AWS CDK Confluence DynamoDB GitHub GitHub Actions JIRA Microservices NestJS Python Terraform TypeScript
5 hours, 19 minutes ago
Agiloft

AI Ops Engineer GTM

Agiloft 51-250 Capital Markets

Agiloft is hiring an AI Ops Engineer, GTM to shape and deliver practical AI solutions across Sales, Marketing, Customer Success, and Revenue Operations, using customer and market insight to improve productivity, revenue outcomes, and customer experience.

United States Full-time Junior AI Engineer Sales Operations
CRM HubSpot Python Salesforce Snowflake
5 hours, 19 minutes ago
CI&T

[Job-31694] Tech Lead AI Engineer | Agentic SDLC, Brazil

CI&T 5K-10K Internet Software & Services

A CI&T busca uma pessoa Tech Lead – AI Engineer/Orchestrator para liderar tecnicamente soluções de software em uma operação estratégica do setor financeiro, conectando arquitetura, desenvolvimento e agentes de IA em um ambiente de Agentic SDLC.

Brazil Full-time Lead AI Engineer Technical Lead
API Gateway AWS CI/CD DevSecOps Generative AI Java Microservices REST API
5 hours, 19 minutes ago
Pavago

AI Systems Engineer (LLM & AI Agent Development)

Pavago IT Services

The company is seeking a remote AI Systems Engineer to build, integrate, and scale production AI applications, autonomous agents, and internal software systems in collaboration with the founder.

Philippines Full-time Senior AI Engineer
HubSpot Java Node.js Python Salesforce
5 hours, 34 minutes ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers