Principal AI Model Architect, Austin, Texas

This is not a remote work opportunity

The Rarest Role in 5G Silicon. Built for the Rarest Kind of Engineer.
The Mission
Most AI runs on someone else's hardware. You're going to design the hardware around the AI.

As the Principal AI Model Architect, you sit at the most critical intersection in our entire silicon program - the place where bleeding-edge AI research collides with custom ASIC reality. You are the "Voice of the Model" - the person who takes ambitious algorithms born in a research lab and answers the hardest question in the building:

"How do we actually run this on silicon we haven't built yet - and make it faster than anyone else on the planet?"

You won't just map models to hardware. You'll shatter them - strategically, surgically - across a heterogeneous ASIC architecture purpose-built for the demands of next-generation 5G AI. Your decisions will be etched into silicon. Your projections will determine tape-out confidence. Your golden models will be the source of truth that an entire SDK team builds against.

This is not a support role. This is the role that makes the chip worth building.

What You'll Own
The Hardware-Software Bridge
You are the translator between two worlds that rarely speak the same language. You'll take complex AI and RAN algorithms - Transformers, Grouped Query Attention, Positional Embeddings - and forge them into concrete hardware specifications for the ASIC team and precise lowering requirements for the SDK team. When there's ambiguity between what research wants and what silicon can deliver, you resolve it.

Model Partitioning & Tiling Strategy
Massive parameter models don't fit neatly onto a chip. You'll define exactly how they do - determining the partitioning, tiling, and memory staging strategies that make heterogeneous execution not just possible, but optimal. This is where deep theoretical knowledge meets brutal real-world constraint, and where the best architects separate themselves from everyone else.

Golden Model Ownership
You maintain the Source of Truth - reference implementations in JAX/PyTorch that capture the researcher's intent with zero ambiguity. You'll work shoulder-to-shoulder with the SDK team to ensure that what comes out of the MLIR compiler is exactly what went into the model. 100% fidelity. No exceptions.

Cycle-Accurate Performance Projection
Before a single gate is synthesized, you'll use your architectural intuition and simulation expertise to project exactly how next-generation silicon targets will perform. Your numbers will drive investment decisions, microarchitecture tradeoffs, and tape-out timelines. You're not guessing - you're proving it before it exists.

Join our Team

What You Bring
Model Architecture Mastery
You have deep, battle-tested knowledge of Transformer architectures, modern attention mechanisms (GQA, MQA), and embedding strategies - not just how they work, but why they work, and exactly where they break under hardware constraints.

Hardware-Aware ML - For Real
You've done Hardware-in-the-Loop research. You think about cache lines, memory walls, HBM vs. SRAM bandwidth ceilings, and SIMD/VLIW execution units while you design models - not after. The memory wall isn't a problem you hand off. It's a design constraint you architect around.

Framework Mastery at the Metal
Advanced proficiency in JAX (strongly preferred), PyTorch, or TensorFlow - specifically in model export, graph capture, and the XLA/StableHLO stack. You know what happens to a model between Python and an execution kernel, and you can control it.

Performance Modeling Depth
Experience with SystemC, TLM, or custom cycle-accurate simulators. You validate performance before tape-out - because after is too late and too expensive.

The Profile That Makes Us Stop Scrolling
Telecom DNA: You've applied AI to real RAN workloads - beamforming, channel estimation, L1/L2/L3 signal processing. You understand that 5G AI is not the same as data center AI.
Compiler Fluency: You don't need to write MLIR passes from scratch - but you absolutely know the difference between a high-level computation graph and a linearized execution schedule, and you can have an intelligent conversation about what happens in between.
Numeric Instinct: You've worked with complex-number representations in AI models and understand how they map to specialized DSP and matrix hardware. You know why this matters and where it breaks.

If this is the role you've been waiting for, we want to hear from you.

What happens once you apply?
Click Here to find all you need to know about what our typical hiring process looks like.

DISCLAIMER: The above statements are intended to describe the general nature and level of work being performed by employees in this position. They are not an exhaustive list of all responsibilities, duties and skills required for this position, and you may be required to perform additional job tasks as assigned.

Primary country and city: USA || Austin, Texas

Job details: Developer

Primary Recruiter: Jim Everett

Compensation and Benefits at Ericsson
At Ericsson, we know that our people are the key to our success. We offer a competitive package to help with your individual needs and goals.

Your Pay
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.

• LOCATION: Austin, Texas

Your Health
Ericsson offers excellent health benefits including the choice of three medical plan options and a dental plan option that allow an employee to select the level of coverage that suits their needs. Employees will receive company credits in an amount equal to the cost that Ericsson pays toward the cost of their medical and dental premiums for themselves and eligible covered dependents.

Your Financial Security

We invest in both your short and long-term financial wellbeing. The Ericsson US 401(k) Plan offers an automatic 3% company contribution and Ericsson match $1 for every $1 you put into the 401(k) Plan on the first 3% of your eligible pay, plus 50 cents on every $1 on the next 2% of eligible pay. When you contribute at least 5% of eligible pay, you are receiving Ericsson's full matching contributions of 4%. Matching and company automatic contributions stop when your total eligible pay for the year reaches the IRS limits. Employees will also receive company credits in an amount equal to the cost of basic life insurance and basic accidental death and dismemberment coverage, as well as short-term and long-term disability coverage. Employees also have the option to participate in Ericsson's Stock Purchase Plan.

Your Time
Your work-life balance is important to us. New employees are provided a minimum of 15 days of accrued vacation, up to 3 personal days per year, 11 annual holidays, 8 hours of volunteer time, and 80 hours of sick time annually. Please note paid time off is pro-rated based on the employee's start date. Furthermore, Ericsson provides up to 16 weeks of paid maternity leave and 6 weeks of parental or adoption leave at 100% of pay.

Additional Benefits
Ericsson offers many other company-paid benefits such as financial wellness programs, educational assistance, matching gifts, and recognition programs.

Ericsson

40 Job Openings at Ericsson

Similar Companies