Lead RL Infra Engineer - Scalable GPU Training Platforms

Advanced Micro Devices; Santa Clara (CA); On-site; USD 120,000 - 170,000; Full time

Job summary

Advanced Micro Devices is hiring a Lead AI Infrastructure Engineer specializing in reinforcement learning. This role involves overseeing RL infrastructure, enhancing productivity for researchers through robust systems and processes.

The ideal candidate should have a Bachelor’s in Computer Science (Master’s or PhD preferred) and experience with machine learning platforms and distributed training. Join AMD to push the boundaries of AI and innovation.

Qualifications

• Bachelor’s degree required; Master’s or PhD preferred.
• Proficiency in C++/Python performance tuning and I/O optimization.
• Experience with distributed training and GPU orchestration.

Responsibilities

• Design and implement distributed RL training stacks.
• Build high-throughput rollout workers and reward computation pipelines.
• Collaborate with research scientists on hyperparameter sweeps.

Skills

Machine learning platforms

PyTorch

C++

Python

GPU cluster orchestration

Education

Bachelor's degree in Computer Science

Master's or PhD in Computer Science

Tools

NCCL

MPI

Job description

Advanced Micro Devices is hiring a Lead AI Infrastructure Engineer specializing in reinforcement learning. This role involves overseeing RL infrastructure, enhancing productivity for researchers through robust systems and processes.

The ideal candidate should have a Bachelor’s in Computer Science (Master’s or PhD preferred) and experience with machine learning platforms and distributed training. Join AMD to push the boundaries of AI and innovation.