Senior AI Compiler Engineer

Ericsson • Austin, TX

6d • Onsite

About The Position

This role focuses on optimizing AI models for future, unreleased silicon, specifically bare-metal 5G/6G telecommunications hardware. The Senior AI Compiler Engineer will build an MLIR-based compiler stack from scratch to run large AI models like Llama 3 on this specialized hardware, where memory is strictly managed and there is no operating system. The position involves closing the gap between research models and hardware-optimized execution on custom silicon.

Requirements

• MLIR Mastery: experience building custom passes, dialect conversions, writing TableGen definitions, and navigating MLIR in a production context.
• Compiler Fundamentals: Deep, hands-on C++ and LLVM expertise, including IR structure, pass pipelines, and code generation pipelines.
• Framework Integration: Experience exporting and lowering models from JAX, PyTorch, or TensorFlow.
• Dialect Expertise: Knowledge of StableHLO or XLA at the op-semantics level.

Nice To Haves

• Experience targeting Embedded DSP or VLIW architectures.
• Advanced optimization depth: loop tiling, vectorization, quantization lowering, fixed-point arithmetic.
• Hardware-level awareness of DMA engines, multi-level memory hierarchies, and complex number handling in MLIR.

Responsibilities

• Design and implement custom MLIR dialects, including TableGen definitions, transformation passes, and dialect conversion infrastructure.
• Lower high-level computational graphs from JAX and StableHLO into hardware-specific IR without loss of mathematical fidelity.
• Build static scheduling and tiling passes to manage data movement between HBM and on-chip local memory with precision.
• Maintain a high-performance, modular, LLVM-based compiler codebase.

Benefits

• Choice of three medical plan options
• Dental plan option
• Company credits towards medical and dental premiums
• 401(k) Plan with automatic 3% company contribution
• Ericsson match $1 for every $1 on the first 3% of eligible pay, plus 50 cents on every $1 on the next 2% of eligible pay (up to 4% match)
• Company credits for basic life insurance and basic accidental death and dismemberment coverage
• Company credits for short-term and long-term disability coverage
• Option to participate in Ericsson’s Stock Purchase Plan
• 15 days of accrued vacation
• Up to 3 personal days per year
• 11 annual holidays
• 8 hours of volunteer time
• 80 hours of sick time annually
• Up to 16 weeks of paid maternity leave
• 6 weeks of parental or adoption leave at 100% pay
• Financial wellness programs
• Educational assistance
• Matching gifts
• Recognition programs

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed