01794f13-11a TPU Kernel Engineer As a TPU Kernel Engineer at Anthropic, you'll be responsible for identifying and addressing performance issues across many different ML systems, including research, training, and inference. A significant portion of this work will involve designing and optimizing kernels for the TPU. You will also provide feedback to researchers about how model changes impact performance.

Strong candidates will have a track record of solving large-scale systems problems and low-level optimization. They should have significant experience optimizing ML systems for TPUs, GPUs, or other accelerators, and be results-oriented with a bias towards flexibility and impact.

Responsibilities:

  • Identify and address performance issues across multiple ML systems
  • Design and optimize kernels for the TPU
  • Provide feedback to researchers on model changes and their impact on performance

Requirements:

  • Bachelor's degree or equivalent combination of education, training, and/or experience
  • Relevant field of study
  • Years of experience required will correlate with the internal job level requirements for the position

Benefits:

  • Competitive compensation and benefits
  • Optional equity donation matching
  • Generous vacation and parental leave
  • Flexible working hours
  • Lovely office space in which to collaborate with colleagues

Note: This job description is a rewritten version of the original ad, focusing on the key responsibilities, requirements, and benefits.

XML job scraping automation by YubHub

]]> full-time mid hybrid $280,000-$850,000 USD ML systems optimization, TPU kernel design and optimization, Large-scale systems problem-solving, Low-level optimization, Results-oriented approach, High-performance computing, Machine learning framework internals, Language modeling with transformers, Accelerator architecture, Collective communication algorithms Engineering Technology Anthropic https://logos.yubhub.co/anthropic.com.png Anthropic creates reliable, interpretable, and steerable AI systems. It is a public benefit corporation headquartered in San Francisco. https://www.anthropic.com/ https://job-boards.greenhouse.io/anthropic/jobs/4720576008 San Francisco, CA | New York City, NY | Seattle, WA 2026-04-18