Lead AI Infrastructure & Accelerator Architect

EPAM Systems Inc; Seattle (WA); Remote; USD 180,000 - 220,000; Full time

Benefits offered by this job

Medical, Dental and Vision Insurance (

Health Savings Account

401(k) Matching

Paid Time Off

Employee Stock Purchase Program

LinkedIn Learning access

Discretionary bonus program

Job summary

EPAM Systems, Inc. seeks a Principal Architect to lead the strategy for high‑performance AI infrastructure, optimization across AI accelerators, and hardware/software co‑design at scale.

You will define the architectural vision for AI training and serving, guide a Center of Excellence, and collaborate with researchers and compiler teams to push performance boundaries for global customers.

Qualifications

• Bachelor's degree or higher in CS/EE or equivalent practical experience

Responsibilities

• Define and drive the multi-year technical roadmap for high-performance AI kernels and hardware-software co-design
• Act as the principal technical liaison with ML researchers and compiler teams to eliminate bottlenecks
• Architect foundational infrastructure including benchmarking suites and autotuning frameworks
• Anticipate industry shifts by tracking advancements in hardware architectures and compiler innovations

Skills

Distributed systems

AI infrastructure

High-performance computing

C++

Python

Leadership

Education

Bachelor's degree in CS/EE (Master's or PhD preferred)

Tools

JAX

PyTorch

XLA

MLIR

Job description

EPAM Systems, Inc. seeks a Principal Architect to lead the strategy for high‑performance AI infrastructure, optimization across AI accelerators, and hardware/software co‑design at scale.

You will define the architectural vision for AI training and serving, guide a Center of Excellence, and collaborate with researchers and compiler teams to push performance boundaries for global customers.