About the Role
We are seeking an AI Compiler Design Engineer to create compilers for specialized AI workloads, optimizing models for future silicon. You will build a complete MLIR-based compiler stack from the ground up to run massive AI models on bare-metal 5G/6G telecommunications silicon without an OS.
Responsibilities
- Define the framework and data flow for building an optimized compiler at the lowest and higher levels.
- Build a complete MLIR-based compiler stack from the ground up for AI models on bare-metal 5G/6G telecommunications silicon.
- 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 losing mathematical fidelity.
- Build static scheduling and tiling passes to manage data movement between HBM and on-chip local memory.
- Maintain a high-performance, modular, LLVM-based compiler codebase.
Requirements
- Experience building MLIR-based compiler stacks.
- Experience with JAX and StableHLO.
- Experience with LLVM.
- Experience with C++.
Skills
- MLIR
- TableGen
- JAX
- StableHLO
- LLVM
- C++
- Compiler Design
- AI Workloads
- 5G/6G Telecommunications Silicon
- Bare-metal programming
- Concurrency Modeling
- Memory Modeling
- Static Scheduling
- Tiling Passes
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.
Benefits
- Opportunity for an annual bonus based on business and individual performance.
- Choice of three medical plan options.
- Dental plan option.
- Company credits towards medical and dental premiums.
- Ericsson US 401(k) Plan with automatic 3% company contribution.
- Ericsson match $1 for $1 on the first 3% of eligible pay and 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.
- 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% 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.
- Ericsson believes this approach drives innovation, which is essential for future growth.
- Ericsson is proud to be an Equal Opportunity employer.
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.
