About the Role
As our Senior AI Compiler Engineer, you will build a complete MLIR-based compiler stack from the ground up, purpose-built to run massive AI models like Llama 3 on bare-metal 5G/6G telecommunications silicon, optimizing for hardware that does not yet exist.
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 for data movement between HBM and on-chip local memory.
- Maintain a high-performance, modular, LLVM-based compiler codebase.
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 real models from JAX, PyTorch, or TensorFlow.
- Dialect Expertise: Knowledge of StableHLO or XLA at the op-semantics level.
Skills
- MLIR Architecture
- Compiler Infrastructure
- 5G/6G Silicon
- Bare-Metal AI
- C++
- LLVM
- JAX
- StableHLO
- PyTorch
- TensorFlow
- Embedded DSP architectures
- VLIW architectures
- Instruction scheduling
- Loop tiling
- Vectorization
- Quantization lowering
- Fixed-point arithmetic
- DMA engines
- Multi-level memory hierarchies
- Complex number handling
Location
- Austin, Texas
Work Type
- Onsite
- Full-time
Experience Level
- Senior
Salary/Compensations
- The salary range for this position is dependent on various factors including, but not limited to, location, and the candidate’s combination of job-related knowledge, qualifications, skills, education, training, and experience
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 and matching contributions
- Company credits for basic life insurance and basic accidental death and dismemberment coverage
- Company credits for short-term and long-term disability coverage
- Optional participation in 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
About the Company
- Ericsson uses a merit-based hiring approach that values people with different experiences, perspectives and skillsets.
- We truly believe this approach drives innovation, which is essential for our future growth.
- Ericsson is proud to be an Equal Opportunity employer.
Equal Opportunity
- Ericsson uses a merit-based hiring approach that values people with different experiences, perspectives and skillsets.
- We truly believe this approach drives innovation, which is essential for our future growth.
- We encourage people from all backgrounds to apply and realize their full potential as part of our Ericsson team.
- Ericsson is proud to be an Equal Opportunity employer, learn more.
- If you need assistance or to request an accommodation due to a disability, please contact Ericsson at hr.direct.americas@ericsson.com.
