Principal AI Performance & Reliability Engineer

Advanced Micro Devices, Inc.; San Jose (CA); Hybrid; USD 250,000 - 420,000; Full time

Benefits offered by this job

Hybrid work model

Benefits package

Job summary

Advanced Micro Devices, Inc. is seeking a Principal or Fellow level software engineer to advance AI infrastructure, focusing on performance, reliability and scalable workloads for model training and inference.

You will collaborate with customers and cross‑functional teams to identify bottlenecks, implement reusable solutions, and shape product capabilities to run demanding AI workloads efficiently at scale.

Qualifications

• PhD or master’s degree in AI, ML, CS or related field or equivalent experience.
• Proven track record in optimizing AI workloads and production systems.
• Experience with large-scale model training and inference pipelines.

Responsibilities

• Profile and optimize AI model training and inference workloads.
• Improve throughput, latency, memory efficiency and scalability.
• Identify bottlenecks across models, frameworks, runtimes and hardware.
• Develop performance tools, benchmarks, automation and observability.
• Collaborate with customers and internal teams to resolve production issues.

Skills

AI model training

Model inference

Performance profiling

Python

C++

PyTorch

TensorFlow

CUDA

Education

PhD or Masters

Tools

PyTorch

TensorFlow

JAX

CUDA

XLA

MLIR

NCCL

Job description

Advanced Micro Devices, Inc. is seeking a Principal or Fellow level software engineer to advance AI infrastructure, focusing on performance, reliability and scalable workloads for model training and inference.

You will collaborate with customers and cross‑functional teams to identify bottlenecks, implement reusable solutions, and shape product capabilities to run demanding AI workloads efficiently at scale.