Senior Staff Engineer - AI Compiler

Ericsson • Austin, TX

1d • Onsite

About The Position

This role focuses on optimizing AI models for custom, bare-metal silicon, specifically 5G/6G telecommunications hardware, rather than existing GPUs. The Senior AI Compiler Engineer will be responsible for building a complete MLIR-based compiler stack from the ground up. This involves tackling the challenge of bridging the gap between research models and hardware-optimized silicon execution, a technically demanding problem in the industry. The position is based in Austin, Texas, and is not a remote opportunity.

Requirements

• MLIR Architecture experience
• Compiler Infrastructure experience
• Experience with 5G/6G Silicon
• Bare-Metal AI experience
• Experience building a complete MLIR-based compiler stack from the ground up.
• Experience lowering computational graphs from JAX and StableHLO.
• Experience with concurrency and memory modeling for bare-metal systems.
• Production C++ experience.
• Experience with LLVM-based compiler codebase.

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 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
• 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-paid basic life insurance
• Company-paid basic accidental death and dismemberment coverage
• Company-paid short-term disability coverage
• Company-paid 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% of 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