Heterogeneous AI Compiler Engineer

Acceler8 Talent; San Francisco (CA); Híbrido; USD 210 000 - 320 000; Tempo integral

Resumo da oferta

Acceler8 Talent is seeking a Member of Technical Staff focused on compilers to help build the AI compiler stack that targets CPUs, GPUs, and accelerators. The role sits at the center of the inference stack, enabling cross-platform scheduling, memory movement, and kernel execution across diverse hardware.

You will design MLIR dialects, optimization passes, and lowering paths, contributing to high-performance code generation and end-to-end inference efficiency on a growing, multi-target compute

Qualificações

• Experience building ML compilers or runtimes with a strong focus on performance optimization.
• Experience with MLIR, LLVM, or comparable compiler infrastructure.
• Experience designing intermediate representations.
• Experience writing compiler passes.
• Experience implementing lowering and code-generation pipelines.
• Strong C++ and/or Python skills.
• Strong understanding of SSA, memory systems, scheduling, and hardware efficiency.
• Experience with operator fusion, tiling, layout transformations, or memory planning.

Responsabilidades

• Scale compiler architecture across a growing set of heterogeneous compute targets.
• Design abstractions, MLIR dialects, compiler passes, and lowering paths for new CPUs/GPUs/accelerators.
• Generate high-performance code competitive with native hardware toolchains.
• Build target-agnostic and target-specific compiler optimizations.
• Implement operator fusion, tiling, layout transformations, memory planning, and execution-planning optimizations.
• Optimize how workloads are partitioned across devices and how intermediate state moves between them.
• Improve end-to-end latency, throughput, utilization, and serving cost.
• Evaluate and integrate automatically generated kernels into compiler code-generation strategies.
• Develop systems to choose between generated kernels, library implementations, and hand-written kernels based on performance.
• Enable new model architectures and inference techniques across multiple hardware targets.
• Work closely with kernel, runtime, ML systems, distributed-systems, and infrastructure engineers.

Conhecimentos

ML compiler experience

MLIR/LLVM

C++

Python

Performance optimization

SSA understanding

Operator fusion/tiling

Profiling performance

Ferramentas

LLVM tooling

MLIR

Profiling tools

Descrição da oferta de emprego

Acceler8 Talent is seeking a Member of Technical Staff focused on compilers to help build the AI compiler stack that targets CPUs, GPUs, and accelerators. The role sits at the center of the inference stack, enabling cross-platform scheduling, memory movement, and kernel execution across diverse hardware.

You will design MLIR dialects, optimization passes, and lowering paths, contributing to high-performance code generation and end-to-end inference efficiency on a growing, multi-target compute