Jobs · Engineering · California

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

AMD · Santa Clara, CA · 3 mo ago
HybridEngineeringFull-time

Key Responsibilities

  • 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

Preferred Experience

  • 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

Benefits

Benefits offered are described: AMD benefits at a glance.

Similar jobs