Compiler Engineer – Machine Learning Compiler

Job not on LinkedIn

🕒 March 19

🏄 California – Remote

info

⏰ Full Time

🟡 Mid-level

🟠 Senior

🤖 Machine Learning Engineer

🦅 H1B Visa Sponsor

info
Apply Now
Find Similar Remote Jobs

📊 Check your resume score for this job

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

Logo of Mythic

Mythic

11 - 50 employees

Founded 2012

🎖️ Defense

🏭 Manufacturing

🤖 Artificial Intelligence

💰 $70M Series C on 2021-05

Defense • Manufacturing • Artificial Intelligence

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

• Join us in building the next generation of AI compilers. You’ll play a key role in developing the compiler for our novel AI accelerator, working side-by-side with hardware engineers and ML researchers. • Your work will shape how deep learning workloads run on cutting-edge dataflow hardware—defining the instruction set, execution model, and developer experience. • The result: a compiler that delivers breakthrough performance while remaining seamless and intuitive for ML developers. • Contribute across the full compiler stack, including operator lowering, graph/IR transformations, optimization passes, and backend code generation • Optimize for dataflow architectures, developing pipelined schedules, memory orchestration, and resource-constrained execution strategies • Collaborate with hardware architects to influence architectural features, ensuring the compiler and hardware evolve together • Develop compilation strategies that unify our analog compute with digital subsystems • Build and maintain a compiler that produces high-performance binaries with strong debugging support, clear error messages, and predictable performance models

🎯 Requirements

• 3+ years of experience building compilers or high-performance systems software, especially those involving complex resource management or optimization. • Expert in modern C++ (C++14/17/20) and strong Python. • Experience with compiler IRs (SSA-based or graph-based), transformations, and code generation • Exposure to specialized accelerators (GPU, NPU, FPGA, or custom ASIC) or parallel architectures • Experience with machine learning compiler stacks (e.g., ONNX, MLIR, TVM, XLA, IREE, PyTorch), with contributions to MLIR or LLVM projects a plus • Experience with optimization methods (LP/MIP, CP, SAT/SMT) using solvers like Gurobi or OR-Tools for scheduling and resource allocation • Experience compiling for specialized accelerators (GPU, NPU, FPGA, or custom ASIC) on DNN workloads; GPU/DSP experience is valuable if combined with compiler backend work beyond kernel tuning • Familiarity with heterogeneous compilation, especially mixing custom accelerators with CPUs/GPUs/NPUs, and exposure to analog or in-memory compute is a plus • Experience collaborating in compiler–hardware co-design (architecture + ISA) for better compiler usability and hardware efficiency

🏖️ Benefits

• Competitive compensation, equity, and benefits package • A collaborative, innovative team that values engineering rigor, continuous integration, and user-focused design. We foster an environment of shared learning and technical excellence • The opportunity to shape how deep learning and LLM workloads are compiled on novel hardware. • A role that spans software and hardware co-design, shaping both the compiler and the accelerator architecture

Apply Now

Similar Jobs

🕒 March 18

Kled AI

11 - 50

🤖 Artificial Intelligence

🏪 Marketplace

🤝 B2B

ML Engineer focusing on fraud detection and data quality for a consumer application. Building large-scale adversarial ML systems with millions of uploads daily.

🇺🇸 United States – Remote

💵 $150k - $250k / year

⏰ Full Time

🟡 Mid-level

🟠 Senior

🤖 Machine Learning Engineer

🕒 March 17

Cresta

51 - 200

☁️ SaaS

🤖 Artificial Intelligence

🏢 Enterprise

Senior Machine Learning Engineer developing advanced ASR systems for Cresta. Focus on model evaluation, quality improvements, and driving measurable business impact in AI-driven experiences.

PyTorch

Tensorflow

🕒 March 13

Button

51 - 200

💼 Consulting

📣 Marketing

☁️ SaaS

Senior Machine Learning Engineer responsible for the full ML lifecycle at Button. Collaborating with data teams to build systems that integrate machine learning into commerce and monetization products.

🕒 March 5

Cantina

51 - 200

🤖 Artificial Intelligence

🎮 Gaming

🌍 Social Impact

Senior Machine Learning Engineer developing AI image generation models for a social AI company focused on enhancing creativity and connections. Join a team revolutionizing online interaction through AI.

🇺🇸 United States – Remote

💵 $200k - $265k / year

⏰ Full Time

🟡 Mid-level

🟠 Senior

🤖 Machine Learning Engineer

🦅 H1B Visa Sponsor

info

🕒 March 4

C the Signs

51 - 200

🏥 Healthcare

⚕️ Healthcare Insurance

🤖 Artificial Intelligence

Senior MLOps Engineer designing and operating ML platforms for healthcare workflows. Working closely with ML, data, security, and product teams, improving model development and deployment.

🇺🇸 United States – Remote

⏰ Full Time

🟠 Senior

🤖 Machine Learning Engineer