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.
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
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
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
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
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.