AI Software Optimization Lead for Large-Scale Models

Advanced Micro Devices, Inc.; Austin (TX); On-site; USD 180,000 - 280,000; Full time

Job summary

Advanced Micro Devices, Inc. is seeking a visionary technical leader to define and drive the end-to-end software optimization strategy for AI workloads.

You will work across architecture, customer engagement, and software engineering to extract maximum performance from AMD hardware. You’ll lead profiling and tuning of large-scale models, partner with hyperscalers, and influence future silicon features through cross-functional collaboration.

Qualifications

• Leadership in a high-impact software environment.
• 0
• Experience solving complex performance bottlenecks at scale.
• Strong knowledge of AI hardware and software optimization.
• Ability to map models to low-level software and roadmaps.

Responsibilities

• Define and drive end-to-end software optimization strategy for AI workloads.
• Lead workload profiling, analysis, and tuning for large models.
• Engage with top customers to understand workload requirements.
• Collaborate with hardware, compiler, and framework teams on silicon features.
• Develop tools for performance estimation, modeling, and reporting.
• Mentor and inspire AMD engineers and external tech communities.

Skills

Strategic leadership

Performance optimization

Cross-functional collaboration

Strong communication

Distributed AI

Education

Master’s or Bachelor’s in CS/EE

Tools

ROCm

PyTorch

JAX

vLLM

SGLang

TorchProfiler

ROCm Profiler

Nsight

Job description

Advanced Micro Devices, Inc. is seeking a visionary technical leader to define and drive the end-to-end software optimization strategy for AI workloads.

You will work across architecture, customer engagement, and software engineering to extract maximum performance from AMD hardware. You’ll lead profiling and tuning of large-scale models, partner with hyperscalers, and influence future silicon features through cross-functional collaboration.