Skip to content
AnthropicAnthropicSan Francisco, CA

Research Engineer, Performance RL

Research Engineer on the Code RL team advancing AI models' ability to write efficient code for accelerators. Requires deep expertise in accelerators like CUDA/ROCm and ML frameworks like JAX/PyTorch, plus experience across kernels, model code, and distributed systems.

350k – 850k/yr
HybridML Engineering

About the role

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.

Requirements

  • Expertise with accelerators (CUDA, ROCm, Triton, Pallas), ML framework programming (JAX or PyTorch).
  • Worked across the stack – kernels, model code, distributed systems.
  • Know how to balance research exploration with engineering implementation.
  • Passionate about AI's potential and committed to developing safe and beneficial systems.

Nice-to-Haves

  • Experience with reinforcement learning.
  • Experience porting ML workloads between different types of accelerators.
  • Familiarity with LLM training methodologies.

Compensation

Annual Salary: $350,000—$850,000 USD

Skills

CUDARocmTritonPallasJAXPyTorchReinforcement LearningDistributed SystemsLlm TrainingKernels

Similar roles

ML Engineering jobs
Thinking Machines Lab

Research Infrastructure Engineer

Thinking Machines LabSan Francisco, CA

Build and operate research infrastructure like evaluation frameworks, RL training systems, and experiment tracking platforms. Partner directly with ML researchers to identify bottlenecks, ensure high adoption of tools, and accelerate research velocity.

350k – 475k/yr
On-siteML Engineering
Anthropic

Research Engineer, Safeguards Labs

AnthropicSan Francisco, CA +1

Research engineer on the Safeguards Labs team building and evaluating novel safety methods to detect misuse, strengthen model safeguards, and reduce real-world harm from Claude.

350k – 850k/yr
HybridML Engineering
Anthropic

Research Engineer, Knowledge Team

AnthropicSan Francisco, CA +2

Designs new information architectures for LLMs to interact with external data sources, implements finetuning/RL training, builds evaluation sets, and develops agentic search capabilities. Requires strong Python/ML skills and LLM experience.

350k – 850k/yr
HybridML Engineering
Thinking Machines Lab

Research, Vision Expertise

Thinking Machines LabSan Francisco, CA

Conducts research on visual perception, multimodal learning, and large-scale AI model training. Designs architectures, builds datasets and evaluations, and collaborates on frontier models. Requires ML expertise, Python proficiency, and experimental rigor.

350k – 475k/yr
On-siteML Engineering
Thinking Machines Lab

Research, Pre-Training Data

Thinking Machines LabSan Francisco, CA

Designs and implements methods for sourcing, curating, and analyzing large-scale pre-training datasets for AI models, blending research with production-grade data engineering. Requires Python proficiency, deep learning frameworks, and strong ML fundamentals.

350k – 475k/yr
On-siteML Engineering