Silicon-Driven AI Architect — Shape Models in Hardware

Ericsson GmbH; Austin (TX); On-site; USD 150,000 - 200,000; Full time

Benefits offered by this job

Competitive salary

Health benefits

401(k) plan with matching

Paid time off and parental leave

Job summary

Ericsson GmbH in Austin, Texas is seeking a Principal AI Model Architect to revolutionize the integration of AI models with silicon hardware. In this pivotal role, you'll leverage your expertise to translate cutting-edge AI algorithms into precise hardware specifications, ensuring seamless functionality.

Candidates should possess deep knowledge of Transformer architectures, as well as proficiency in JAX and PyTorch. The position offers a comprehensive benefits package, including competitive salary, health benefits, and vacation time.

Qualifications

• Deep knowledge of Transformer architectures and design choices.
• Experience in hardware-aware machine learning.
• Advanced proficiency in JAX, PyTorch, or TensorFlow.

Responsibilities

• Translate AI algorithms into hardware specifications.
• Define strategies for model partitioning and tiling.
• Maintain canonical reference implementations in major ML frameworks.

Skills

Model Architecture Mastery

Hardware‑Aware ML

Framework Depth

Performance Modeling

Education

Advanced degree in relevant field

Tools

JAX

PyTorch

TensorFlow

SystemC

Job description

Ericsson GmbH in Austin, Texas is seeking a Principal AI Model Architect to revolutionize the integration of AI models with silicon hardware. In this pivotal role, you'll leverage your expertise to translate cutting-edge AI algorithms into precise hardware specifications, ensuring seamless functionality.

Candidates should possess deep knowledge of Transformer architectures, as well as proficiency in JAX and PyTorch. The position offers a comprehensive benefits package, including competitive salary, health benefits, and vacation time.