Senior AI Performance & Reliability Engineer

AMD; San Jose (CA); Hybrid; USD 180,000 - 260,000; Full time

Job summary

AMD is seeking a Principal/Fellow level software engineer for the AI Infrastructure team in San Jose/Bellevue area. You will optimize AI workloads (training and inference) across models, frameworks, and hardware, working with customers to identify bottlenecks and deliver scalable solutions.

The role requires strong software engineering, profiling, and experience with ML frameworks like PyTorch, TensorFlow, or JAX.

Qualifications

• Profile and optimize AI model training and inference workloads.
• Improve model throughput, latency, memory efficiency, scalability, and reliability.
• Identify bottlenecks across models, frameworks, compilers, runtimes, OS, and hardware.
• Collaborate with customers to understand requirements and reproduce issues.

Responsibilities

• Develop performance tooling, benchmarks, automation, and observability systems.
• Translate customer feedback into product and infrastructure improvements.
• Work with ML engineers, systems engineers, hardware teams, and product teams.

Skills

Python

C++

Performance profiling

ML frameworks

Education

PhD in AI/CS or related field

Tools

CUDA

ROCm/HIP

NCCL

XLA/MLIR

Job description

AMD is seeking a Principal/Fellow level software engineer for the AI Infrastructure team in San Jose/Bellevue area. You will optimize AI workloads (training and inference) across models, frameworks, and hardware, working with customers to identify bottlenecks and deliver scalable solutions.

The role requires strong software engineering, profiling, and experience with ML frameworks like PyTorch, TensorFlow, or JAX.