Haystack
← Back to Jobs
Technology

machine learning compiler engineer

Ab Ovo IncUnited States🇺🇸United StatesPosted 20 Jul 2026

Why This Role Stands Out

This hybrid role offers an exciting opportunity to shape the future of Machine Learning compilers, contributing to cutting-edge research and open-source projects while enjoying flexible work arrangements. If you possess a strong foundation in compiler development, ML, and software engineering, and are driven by innovation and intellectual challenge, you'll thrive in this dynamic environment. Dive into a career where you'll make a significant impact and continuously expand your expertise.

Quick Overview

Work Type
Hybrid
Level
Mid Senior

Job Description

Responsibilities

  • Work on a wide range of ML compilers
  • Improve ML compiler optimization capabilities through benchmark analysis and profiling
  • Innovate new ML compiler and optimization algorithms
  • Incorporate generative AI into the compiler development workflow and into compilers themselves
  • Upstream compiler algorithms to open-source compiler projects
  • Author research publications and represent the company in conferences and industry forums
  • For senior levels — lead and manage projects while doing substantial technical work

Required

  • Experience with compiler development and computer architecture
  • ML experience
  • A degree in the field of computer science or applied mathematics
  • Experience with software engineering
  • Solid intellectual ability, motivation, and a strong history of achievement
  • Excellent oral and written communication skills

Desired

  • Experience with MLIR, MLIR dialects (LinAlg, Affine), PyTorch 2.0, TVM, Triton, and/or LLVM
  • Experience incorporating generative AI into compiler development workflow (AI-assisted code generation, AI-assisted analysis, AI-assisted refactoring and review)
  • Experience designing or extending DSLs as targets for AI-driven program synthesis
  • ML applications and ML optimization experience
  • ML architecture experience
  • High-performance computing experience, including communication-avoiding algorithms
  • Polyhedral compiler optimization experience
  • Loop transformation and vectorization experience
  • GPU programming, parallel programming experience
  • General optimization experience

 

Minimum Qualifications:

• Bachelor''s degree in Computer Science, Engineering, Information Systems, or related field and 8+ years of Hardware Engineering, Software Engineering, Systems Engineering, or related work experience.
OR
Master''s degree in Computer Science, Engineering, Information Systems, or related field and 7+ years of Hardware Engineering, Software Engineering, Systems Engineering, or related work experience.
OR
PhD in Computer Science, Engineering, Information Systems, or related field and 6+ years of Hardware Engineering, Software Engineering, Systems Engineering, or related work experience.

 

 

 

 

 

 

Skills

Generative AI
PyTorch

Similar jobs

Browse related jobs

Apply now