About the Role
As a Senior AI Compiler Engineer on the Research Team, you will build a complete MLIR-based compiler stack from the ground up. This stack will be purpose-built to run massive AI models on bare-metal 5G/6G telecommunications silicon, optimizing for environments with limited memory and no operating system.
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 mathematical fidelity loss.
- Build static scheduling and tiling passes to precisely manage data movement between HBM and on-chip local memory.
- Maintain a high-performance, modular, LLVM-based compiler codebase.
- Ensure correctness and determinism in production compiler engineering.
Requirements
- MLIR mastery, including custom passes, dialect conversions, TableGen definitions, and navigating the MLIR framework in a production context.
- Deep, hands-on C++ and LLVM expertise, including IR structure, pass pipelines, and code generation.
- Experience exporting and lowering real models from JAX, PyTorch, or TensorFlow.
- Expertise in StableHLO or XLA at the op-semantics level.
Skills
- MLIR Architecture
- Compiler Infrastructure
- 5G/6G Silicon
- Bare-Metal AI
- C++
- LLVM
- JAX
- PyTorch
- TensorFlow
- StableHLO
- XLA
- Embedded DSP
- VLIW architectures
- Instruction scheduling
- Register pressure
- 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 accidental death and dismemberment coverage
- Short-term and long-term disability coverage
- Optional participation in 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
About the Company
- Ericsson uses a merit-based hiring approach that values people with different experiences, perspectives and skillsets.
- Ericsson truly believes this approach drives innovation, which is essential for our future growth.
- Ericsson provides a competitive package to help with your individual needs and goals.
Equal Opportunity
- 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.
