Senior AI Performance Architect — GPU & Network

Advanced Micro Devices; San Jose (CA); Hybrid; USD 130,000 - 160,000; Full time

Benefits offered by this job

Comprehensive benefits package

Job summary

Advanced Micro Devices is seeking a Senior-level Engineer in San Jose, CA to drive performance optimization strategies for AI on AMD GPUs. This role requires a PhD or master’s degree in a related field, with extensive experience in AI frameworks and network hardware architecture.

The ideal candidate will work collaboratively across teams, focusing on performance analysis, optimization, and strategy formulation. This position offers a hybrid working environment and the chance to shape the future of AI technologies.

Qualifications

• Deep knowledge with network, NIC and GPU hardware architecture.
• Hands-on experience in mapping model architecture to software and hardware.
• Strong knowledge in generative model architecture and distributed inference.

Responsibilities

• Set strategy and roadmap for AMD Collectives and Network optimizations.
• Performance tuning and analysis of large-scale AI models.
• Participate in hardware-software co-design for future optimizations.

Skills

Performance optimization

AI frameworks

Collaboration

Technical leadership

Education

PhD or master's degree in computer science or electrical engineering

Tools

PyTorch

JAX

vLLM

SGLang

Job description

Advanced Micro Devices is seeking a Senior-level Engineer in San Jose, CA to drive performance optimization strategies for AI on AMD GPUs. This role requires a PhD or master’s degree in a related field, with extensive experience in AI frameworks and network hardware architecture.

The ideal candidate will work collaboratively across teams, focusing on performance analysis, optimization, and strategy formulation. This position offers a hybrid working environment and the chance to shape the future of AI technologies.