Principal AI Infrastructure & Accelerator Architect

EPAM Systems

Remote · United StatesSalary not disclosedfull-timeEPAM Systems Careers (remote abroad)

Are you a visionary technical leader passionate about defining the future of AI infrastructure and squeezing every drop of performance out of advanced hardware accelerators at scale? We are seeking a Principal Architect to lead, shape, and execute our technical strategy for AI performance, optimization, and hardware-software co-design.

Responsibilities

Define and drive the multi-year technical roadmap for high-performance AI kernels, custom operations, and hardware-software co-design targeting TPU and GPU architectures
Scale and mentor a world-class technical practice, establishing architectural governance, engineering standards, and best practices across the organization
Act as the principal technical liaison partnering with ML researchers, core framework architects (JAX, PyTorch), and compiler engineering teams (XLA, MLIR) to eliminate systemic bottlenecks and shape future hardware/software requirements
Architect foundational infrastructure—including enterprise-grade benchmarking suites, automated autotuning frameworks, regression analysis pipelines, and comprehensive documentation—empowering the global developer community
Anticipate industry shifts by tracking advancements in hardware architectures, emerging model topologies, and compiler innovations to unlock step-changes in AI training and inference efficiency

Languages

Work format: Remote
Seniority: Lead
Posted: 29 Sept 2026
Last verified: In the last 5 days

Similar roles

Keep looking

More Software jobs in Armenia and remote All IT vacancies Internships and entry level