ML Compiler Engineer, Edge-AI & Low-Latency

femtoAI; San Bruno (CA); On-site; USD 100,000 - 150,000; Full time

Benefits offered by this job

401(k)

Medical insurance

Vision insurance

Disability insurance

Paid maternity leave

Paid paternity leave

Child care support

Job summary

A technology company based in San Bruno is seeking a Compiler Engineer to work on a custom ML compiler for their AI accelerator. The role requires a minimum of 2 years experience in compilers or edge-AI, with strong proficiency in Python and/or C++. You will be responsible for building model ingestion pipelines, implementing graph transformations, and debugging. The position offers benefits including medical insurance, 401(k), and paid leave for parents.

Qualifications

• 2+ years of experience in compilers and/or edge-AI.
• Proficiency in Python and/or C++.
• Ability to reason about correctness and performance tradeoffs.

Responsibilities

• Build and maintain model ingestion pipelines.
• Implement graph transformations such as operator decomposition.
• Debug correctness and numerical issues across transformations.

Skills

Compiler experience

Python proficiency

C++ proficiency

Graph-level transformations

Understanding of deep learning models

Tools

MLIR

LLVM

TVM

XLA

Job description

A technology company based in San Bruno is seeking a Compiler Engineer to work on a custom ML compiler for their AI accelerator. The role requires a minimum of 2 years experience in compilers or edge-AI, with strong proficiency in Python and/or C++. You will be responsible for building model ingestion pipelines, implementing graph transformations, and debugging. The position offers benefits including medical insurance, 401(k), and paid leave for parents.