Principal AI Engineer, Generative Models at Scale

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

Job summary

Advanced Micro Devices, Inc. in San Jose, CA, is seeking a Principal Engineer for Generative AI training and inference at scale.

You will join a core team focused on scalable training/inference across latest AI models, leveraging AMD devices and distributed systems. You will work with PyTorch, JAX and related tools, collaborate with hardware and software teams, and publish ongoing results at conferences while driving end-to-end performance improvements.

Qualifications

• Experience in Generative AI model training and inference.
• Publications in efficient model architectures or parallelism are a plus.
• Proven ability to work with deep learning frameworks and large-scale systems.

Responsibilities

• Propose and apply innovative training and inference techniques for large models.
• Develop scalable architectures and parallelism strategies for AMD platforms.
• Collaborate with open-source communities to integrate AMD-optimized models.
• Co-optimize end-to-end performance with software and hardware teams.
• Publish and present work at external conferences and venues.
• Promote adoption of AI workflows across AMD platforms.

Skills

Generative AI

Distributed training

PyTorch

JAX

vLLM

SGLang

MuJoCo

Education

PhD or Master’s degree in computer science or related field

Tools

PyTorch

JAX

vLLM

SGLang

MuJoCo

Job description

Advanced Micro Devices, Inc. in San Jose, CA, is seeking a Principal Engineer for Generative AI training and inference at scale.

You will join a core team focused on scalable training/inference across latest AI models, leveraging AMD devices and distributed systems. You will work with PyTorch, JAX and related tools, collaborate with hardware and software teams, and publish ongoing results at conferences while driving end-to-end performance improvements.