TPU Kernel Engineer

New York City, NYfull time

On Site

$280k–$850k

H1B: yes

Anthropic TPU Kernel Engineer at Anthropic is responsible for identifying and addressing performance issues in ML systems, focusing on designing and optimizing kernels for TPU. The role includes debugging kernel performance and collaborating with researchers on model performance.

Responsibilities

As a TPU Performance Engineer, 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.
You may be a good fit if you:
Have significant experience optimizing ML systems for TPUs, GPUs, or other accelerators
Are results-oriented, with a bias towards flexibility and impact
Pick up slack, even if it goes outside your job description
Enjoy pair programming (we love to pair!)
Want to learn more about machine learning research
Care about the societal impacts of your work
Strong candidates may also have experience with:
High performance, large-scale ML systems
Designing and implementing kernels for TPUs or other ML accelerators
Understanding accelerators at a deep level, e.g. a background in computer architecture

Qualification

Experience with high performanceBackground in computer architectureExperience with ML framework internals

Required

Bachelor’s degree or equivalent combination of education, training, and/or experience
Experience optimizing ML systems for TPUs, GPUs, or other accelerators
Experience solving large-scale systems problems and low-level optimization

Preferred

Experience with high performance, large-scale ML systems
Experience designing and implementing kernels for ML accelerators
Background in computer architecture
Experience with ML framework internals
Experience with language modeling using transformers

Benefits

Equity/RSU
Parental Leave

Company

Anthropic

View company page

501-1000 employees

Artificial Intelligence

Founded 2021

Source: greenhouse

Active within the past hour

More roles at Anthropic

Product Marketing Manager, Platform

San Francisco, CA

Recruiting Analytics Data Engineer

San Francisco, CA

Research Engineer, Visual Knowledge Work

San Francisco, CA

Startup Partnerships Lead

San Francisco, CA

Boost Your Interview Chances

Improve Resume Match Score

Free

Your Score

9.0

Strong Applicant

Must-Have Skills for This Role

TPUGPUKernel designMachine learningAssemblyLow-precision inference

Optimize my Resume

Apply Faster with Autofill

Free

Save your profile once. LumiJobs pre-fills the employer's application form so you stop re-typing the same answers on every site.

Apply With Autofill

Or skip LumiJobs's tools and submit directly on the employer's career site.

Apply on Employer Site

TPU Kernel Engineer at Anthropic · LumiJobs