TPU Kernel Engineer, Anthropic
$315-560k
+ equity donation matching
More information about location
2+ days a week in office
AI safety and research company
Be an early applicant
AI safety and research company
1001+ employees
Be an early applicant
$315-560k
+ equity donation matching
More information about location
2+ days a week in office
1001+ employees
Company mission
To create reliable, interpretable, and steerable AI systems.
Role
Who you are
- Strong candidates will have a track record of solving large-scale systems problems and low-level optimization
- 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
- ML framework internals
- Language modeling with transformers
- Education requirements: We require at least a Bachelor's degree in a related field or equivalent experience
What the job involves
- As a TPU Kernel 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
- Implement low-latency, high-throughput sampling for large language models
- Adapt existing models for low-precision inference
- Build quantitative models of system performance
- Design and implement custom collective communication algorithms
- Debug kernel performance at the assembly level
Our take
Funded by Google and founded by ex-OpenAI employees, Anthropic is an AI company focused on building a reliable, innovative AI system. While the company is relatively new to the market it has already launched a highly anticipated AI chat assistant called Claude which is set to rival OpenAI's ChatGPT.
Similar to other chatbots, Claude is accessible through a chat interface and can provide a number of conversation and text-processing tasks. A key point of differentiation is that Claude is designed to produce less harmful outputs than that of the other chatbots that came before it.
While there is much praise for the chat assistant, there is still room for improvement with feedback reporting that Claude is worse at maths and grammar compared to its rival. However, with industry giants like Google backing this company from an early stage, it is clear Anthropic has the potential to become a leading player within the AI industry.
Freddie
Company Specialist at Welcome to the Jungle
Insights
190% employee growth in 12 months
Glassdoor (4.3)
Trustpilot (1.8)
Company
Funding (last 2 of 12 rounds)
Mar 2025
$3.5bn
SERIES E
Jan 2025
$1bn
GROWTH EQUITY VC
Total funding: $16.3bn
Company benefits
- Comprehensive health, dental, and vision insurance for you and your dependents
- Inclusive fertility benefits via Carrot Fertility
- Generous subsidy for OneMedical
- 21 weeks of paid parental leave
- Unlimited PTO
- Optional equity donation matching at a 3:1 ratio, up to 50% of your equity grant
- 401(k) plan with 4% matching
- $500/month flexible wellness stipend
- Commuter coverage
- Annual education stipend
- A home office improvement stipend when you first join
- Relocation support for those moving to the Bay Area
Company HQ
SoMa, San Francisco, CA
Leadership
Dario Amodei
(CEO & Co-Founder)Spent almost 5 years at OpenAI as a Team Lead for AI safety and as the Vice President of Research. Also worked as a Research Scientist at Google.
Jared Kaplan
(Chief Science Officer & Co-Founder)Has been a Professor at John Hopkins University for the past 10 years. Also an experienced researcher having worked previously at OpenAI and the SLAC National Accelerator Laboratory.
Jack Clark
(Co-Founder)Experienced reporter having worked previously at the Register and Bloomberg. Also worked for OpenAI in Strategy and Communications and as a Policy Director.
Sam McCandlish
(Co-Founder)Graduated from Stanford University with a PhD in Theoretical Physics. Previously worked as a Research Lead at OpenAI.
Tom Brown
(Co-Founder)Previously worked as a Member of the Technical Staff for both OpenAI and Google Brain.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 166 more jobs at Anthropic