AI Research Scientist, Reinforcement Learning (LLM) and Post-Training

Advanced Micro Devices, Inc.
Santa Clara, CA, United States
about 1 month ago

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Compensation
$204,000.0
Working hours
Regular working hours

Tech stack

Artificial Intelligence Machine Learning Reinforcement Learning Data Logging Graphics Processing Unit (GPU) Large Language Models Information Technology

Job description

We are hiring a Lead AI Research Scientist, Reinforcement Learning (LLM) and Post-Training, specializing in reinforcement learning to advance post-training and interactive learning for large generative models applied to demanding engineering and hardware-adjacent tasks (code, optimization, tool use, and long-horizon decision making). You will invent and analyze RL algorithms-policy optimization, preference-based methods, exploration, credit assignment, and reward modeling-run rigorous empirical studies, and partner with infra and product teams to land methods that improve measurable task success without sacrificing stability or safety., * Research and develop RL methods for post-training LLMs and code models on structured engineering tasks with verifiable or preference-based feedback

  • Design reward models, curricula, and off-policy or on-policy training recipes suited to sparse, noisy, or expensive labels from experts and simulators
  • Characterize failure modes (reward hacking, degenerate policies, instability) and propose mitigations grounded in experiments
  • Collaborate with RL infra engineers to scale training; define interfaces for rollout generation, logging, and reproducibility
  • Publish at top venues (e.g. NeurIPS, ICML, ICLR) and contribute internal technical leadership on the RL roadmap, AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD’s “Responsible AI Policy” is available here.

Requirements

  • Strong publication record in reinforcement learning or closely related machine learning areas.
  • Hands-on experience training RL or preference-optimized models at non-trivial scale (GPUs, distributed jobs)
  • Experience with LLM post-training, RLHF/RLAIF, or policy optimization for language or code agents
  • Familiarity with compilers, kernels, EDA-style workflows, or large-scale codebases is a plus

ACADEMIC CREDENTIALS:

  • PhD in Computer Science, Machine Learning, or related field strongly preferred.

About the company

At AMD, our mission is to build great products that accelerate next-generation computing experiences-from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you’ll discover the real differentiator is our culture. We push the limits of innovation to solve the world’s most important challenges-striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond. Together, we advance your career.

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on jobs.localjobnetwork.com

Good distractions

Talks and stories from around this role — technically off-topic, practically not.

4:26 min

Background on the Data Lab and organizational structure

Álvaro Martín Lozano · LIVE

1:10 min

Exposing sensitive information through partial search logs

Dennis Schulz Dennis Schulz +1 · WWC Europe 2026

2:36 min

Applying supervised machine learning for practical rule extraction

Katja Träumner

3:32 min

Fundamentals and limitations of large language models

Krzystof Czieslak · LIVE

2:01 min

Exploring foundational expertise in traditional optimization and machine learning

Eric Enge · Coffee With Developers

1:56 min

Discovering incidents using logs, metrics, and traces

Nele Uhlemann · WWC 2023

Videos

See all

Related articles

See all