Compiler Developer

Ericsson•Austin, TX

19h•Onsite

About The Position

This is a unique opportunity for an AI Compiler Engineer to join the Research Team and optimize AI models for novel, bare-metal 5G/6G telecommunications silicon that does not yet exist. Unlike typical roles, this position involves building a complete MLIR-based compiler stack from the ground up, specifically designed for massive AI models like Llama 3. The role addresses the challenging gap between research models and hardware-optimized silicon execution, requiring engineers to work without an OS and manage memory meticulously. The engineer will be responsible for closing this gap, contributing to the development of next-generation telecommunications infrastructure.

Requirements

2 plus years of hands-on experience.
Masters Degree in a related discipline.
MLIR Mastery: experience building custom passes and 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 real 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 from scratch, including TableGen definitions, transformation passes, and dialect conversion infrastructure.
Lower high-level computational graphs from JAX and StableHLO into hardware-specific IR without losing mathematical fidelity.
Build static scheduling and tiling passes to choreograph data movement between HBM and on-chip local memory with precision.
Maintain a high-performance, modular, LLVM-based compiler codebase.
Ensure correctness and determinism in production compiler engineering.

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.
401(k) matching: $1 for $1 on the first 3% of eligible pay, plus 50 cents on every $1 on the next 2% of eligible pay.
Company credits towards basic life insurance and basic accidental death and dismemberment coverage.
Company credits towards short-term and long-term disability coverage.
Option to participate in Ericsson’s Stock Purchase Plan.
Minimum of 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.