TPU Kernel Engineer — Lead Low-Latency ML Kernels (Hybrid)

Anthropic; New York (NY); Hybrid; USD 280,000 - 850,000; Full time

Job summary

Anthropic is seeking a TPU Kernel Engineer to identify and tackle performance issues in ML systems including research, training, and inference. The role involves designing and optimizing TPU kernels and providing feedback about the impact of model changes on performance. The ideal candidate has strong experience in optimizing ML systems and is highly results-oriented. The position comes with a competitive salary range of $280,000 to $850,000, and it requires at least a Bachelor’s degree in a relevant field.

Qualifications

• Significant experience optimizing ML systems.
• Strong candidates have a track record of solving large-scale systems problems.

Responsibilities

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

Skills

Optimizing ML systems for TPUs, GPUs, or other accelerators

Flexibility and impact orientation

Pair programming

Interest in machine learning research

Societal impacts awareness

Education

Bachelor’s degree or equivalent

Job description

Anthropic is seeking a TPU Kernel Engineer to identify and tackle performance issues in ML systems including research, training, and inference. The role involves designing and optimizing TPU kernels and providing feedback about the impact of model changes on performance. The ideal candidate has strong experience in optimizing ML systems and is highly results-oriented. The position comes with a competitive salary range of $280,000 to $850,000, and it requires at least a Bachelor’s degree in a relevant field.