jobs

TPU Kernel Engineer

About Anthropic Anthropic’s mission is to create reliable, interpretable, and steerable AI systems. We want AI to be safe and beneficial for our users and for society as a whole. Our team is a quickly growing group of committed researchers,

Company

Anthropic

Location

San Francisco, CA; New York City, NY; Seattle, WA

Status

Open

Posted

2025-05-22T20:19:35+00:00

Anthropic is hiring a TPU Kernel Engineer to identify and fix performance problems in ML systems for research, training and inference. The work involves designing and optimizing TPU kernels, adapting models for low-precision inference, building collective communication algorithms, and debugging kernels at the assembly level. The posting asks for experience optimizing ML systems on TPUs, GPUs or other accelerators, plus large-scale systems work; computer architecture, framework internals and transformer language modelling are listed as extras. Pay range: $280,000 to $850,000, and the team says it pairs often.