RemoteLeaf Logo
Log in Sign up

Compiler Engineer - Machine Learning Compiler

6 months, 4 weeks ago
Anywhere, United States, Canada
Full-time
Mid Level
Machine Learning 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

  • Design and implement components across the full compiler stack, including operator lowering, graph/IR transformations, optimization passes, and backend code generation.
  • Optimize compilation for dataflow architectures by developing pipelined schedules, memory orchestration, and resource-constrained execution strategies.
  • Define and refine the instruction set, execution model, and developer-facing interfaces to shape how workloads run on the accelerator.
  • Collaborate closely with hardware architects, ML researchers, and engineers to influence architectural features and ensure compiler–hardware co-evolution.
  • Develop compilation strategies that unify analog compute with digital subsystems and support heterogeneous execution.
  • Build and maintain a compiler and tooling that produce high-performance binaries with strong debugging support, clear error messages, and predictable performance models.
  • Contribute to continuous integration, testing, and performance validation to ensure reliable and reproducible compiler behavior.

Requirements

  • 3+ years of experience building compilers or high-performance systems software, particularly involving complex resource management or optimization.
  • Expert-level experience with modern C++ (C++14/17/20) and strong proficiency in Python.
  • Experience with compiler IRs (SSA-based or graph-based), transformations, and code generation.
  • Exposure to specialized accelerators or parallel architectures (GPU, NPU, FPGA, or custom ASIC).
  • Experience optimizing for resource-constrained or dataflow-style architectures (scheduling, memory orchestration).
  • Experience with ML compiler stacks (ONNX, MLIR, TVM, XLA, IREE, PyTorch) and/or contributions to MLIR or LLVM projects (nice to have).
  • Familiarity with optimization methods (LP/MIP, CP, SAT/SMT) and solvers like Gurobi or OR-Tools for scheduling/resource allocation (nice to have).
  • Experience compiling DNN workloads for specialized accelerators or GPUs/DSPs with compiler backend work beyond kernel tuning (nice to have).
  • Familiarity with heterogeneous compilation (mixing custom accelerators with CPUs/GPUs/NPUs) and exposure to analog or in-memory compute, and prior compiler–hardware co-design experience (nice to have).

Benefits

  • Competitive compensation with equity package.
  • Comprehensive benefits package.
  • Flexible/remote work options (locations include Palo Alto, Austin, Vancouver, and remote).
  • High-impact role spanning software and hardware co-design with the opportunity to shape how deep learning and LLM workloads are compiled on novel hardware.
  • Collaborative, innovative team culture emphasizing engineering rigor, continuous integration, shared learning, and professional development.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Lyra Health

Manager, AI/ML Engineering

Lyra Health 1K-5K Health Care Providers & Services

Lyra Health is seeking an engineering leader to execute its machine learning roadmap by scaling and mentoring an AI/ML engineering team that delivers reliable, production-grade systems for mental health care.

United States Full-time Lead Engineering Manager Machine Learning Engineer
$161k-$222k
AWS HIPAA Java Kotlin Kubernetes Machine Learning Microservices MLOps Neural Networks Prototyping Transformers
4 hours, 42 minutes ago
Artera

Senior/Staff Machine Learning Engineer (Model Dev)

Artera 51-250 Construction & Engineering

Artera is seeking an experienced machine learning engineer to lead end-to-end AI biomarker development for cancer care, from clinical problem definition and model validation through regulatory submission and production deployment.

United States Full-time Lead Machine Learning Engineer
$180k-$240k
AWS Deep Learning Kubernetes Machine Learning PyTorch TensorFlow
4 hours, 57 minutes ago
CI&T

[Job-31696] Tech Lead – AI Orchestrator | Android, Brazil

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

A CI&T busca uma pessoa Tech Lead – AI Orchestrator | Android para liderar a construção e modernização de jornadas de pagamentos em um SuperApp, integrando fluxos via BFF e orquestrando agentes de IA ao longo do SDLC.

Brazil Full-time Lead Android Engineer Machine Learning Engineer
Android Espresso Generative AI Jetpack Compose JUnit Kotlin REST API XML
1 day, 4 hours ago
Natera

Software Engineer III, Service AI Transformation

Natera 1K-5K Pharmaceuticals

Natera is hiring a Software Engineer III for its Service AI Transformation Team to build reliable, scalable AI-powered services and integrations that streamline clinical, commercial, and operational workflows.

United States Full-time Mid Level Machine Learning Engineer
$106k-$132k
AWS AWS CDK CI/CD Datadog GraphQL HIPAA JavaScript Kafka Microservices Next.js React Serverless Terraform TypeScript
2 days, 4 hours 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