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