AI Compiler Engineer (Hybrid) | Equity Options

Apex Compute; Los Altos, Northern (CA, KY); Hybrid; USD 180,000 - 260,000; Full time

Benefits offered by this job

Stock options

Job summary

Apex Compute is seeking a highly motivated Compiler Engineer to join our Los Altos-based team. This full-time role focuses on developing a compiler stack for our AI compute hardware, with emphasis on lowering ML workloads into MLIR dialects and other IRs, and integrating with ML frameworks.

Ideal candidates will design optimization passes, work with hardware and software teams, and contribute to performance-driven compiler optimizations for AI accelerators.

Qualifications

• Bachelor's or Master's in CS/CE or related field.
• Proven experience with compiler frameworks such as MLIR, LLVM, TVM, XLA, or similar systems.
• Strong understanding of intermediate representations, compiler passes, and optimization pipelines.

Responsibilities

• Design and implement compiler infrastructure and optimization passes mapping ML workloads to architecture primitives.
• Develop lowering pipelines using MLIR, LLVM, or similar frameworks.

Skills

C++

Python

Tensor graphs

Analytical thinking

Education

Bachelor's or Master's in CS/CE

Tools

MLIR

LLVM

TVM

XLA

PyTorch

Job description

Apex Compute is seeking a highly motivated Compiler Engineer to join our Los Altos-based team. This full-time role focuses on developing a compiler stack for our AI compute hardware, with emphasis on lowering ML workloads into MLIR dialects and other IRs, and integrating with ML frameworks.

Ideal candidates will design optimization passes, work with hardware and software teams, and contribute to performance-driven compiler optimizations for AI accelerators.