Anthropic is hiring a Research Engineer in its RL organization to advance the performance of reinforcement learning models for accelerator code generation. The role involves inventing and implementing RL environments, conducting experiments, delivering work into training runs, and collaborating with researchers, engineers, and performance specialists. Candidates should have expertise with accelerators and ML framework programming, experience across the stack, and a passion for safe AI.
Research Engineer, Performance RL (Reinforcement Learning) at Anthropic
Hybrid - San Francisco, CA
More jobs at AnthropicSalary
USD 350,000 - 850,000
Requirements
Skills
- Expertise with accelerators (CUDA, ROCm, Triton, Pallas)
- ML framework programming (JAX or PyTorch)
- Experience across the stack – kernels, model code, distributed systems
- Ability to balance research exploration with engineering implementation
- Passion for AI's potential and commitment to developing safe and beneficial systems
- Bachelor’s degree or equivalent education, training, or experience
- Field relevant to the role (e.g., computer science, machine learning, etc.)
- Experience with reinforcement learning (preferred)
- Experience porting ML workloads between different types of accelerators (preferred)
- Familiarity with LLM training methodologies (preferred)
Responsibilities
- Invent, design, and implement RL environments and evaluations
- Conduct experiments and shape our research roadmap
- Deliver your work into training runs
- Collaborate with other researchers, engineers, and performance engineering specialists across and outside Anthropic
Technologies
CUDAROCmTritonPallasJAXPyTorch
See if your resume is ready for this job
See how our AI can optimize your resume and improve your chances for this role.