Anthropic · 2 weeks ago
TPU Kernel Engineer
Seattle, WA
Full-time
Hybrid
Senior Level
$280K/yr - $560K/yrAnthropic is a public benefit corporation focused on creating reliable and beneficial AI systems. The TPU Kernel Engineer will identify and address performance issues across various ML systems, design and optimize kernels for TPUs, and provide feedback to researchers on model performance impacts.
Artificial Intelligence (AI)Foundational AIGenerative AIInformation TechnologyMachine Learning
Responsibilities
Identifying and addressing performance issues across many different ML systems, including research, training, and inference
Designing and optimizing kernels for the TPU
Providing feedback to researchers about how model changes impact performance
Implementing low-latency, high-throughput sampling for large language models
Adapting existing models for low-precision inference
Building quantitative models of system performance
Designing and implementing custom collective communication algorithms
Debugging kernel performance at the assembly level
Qualification
Required
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
Education requirements: We require at least a Bachelor's degree in a related field or equivalent experience
Preferred
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
Benefits
Equity
Benefits
Incentive compensation
Optional equity donation matching
Generous vacation and parental leave
Flexible working hours
Company
Anthropic
Anthropic is an AI research company that focuses on the safety and alignment of AI systems with human values.
Founded in 2021
501-1000 employees
H1B Sponsorship
Anthropic has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (105)
2024 (13)
2023 (3)
2022 (4)
2021 (1)
Funding
Current Stage
Late StageTotal Funding
$33.74BKey Investors
Lightspeed Venture PartnersGoogleAmazon
2025-09-02Series F· $13B
2025-05-16Debt Financing· $2.5B
2025-03-03Series E· $3.5B
Leadership Team
Dario Amodei
CEO & Co-Founder
Daniela Amodei
President and co-founder
Recent News
2026-01-11
Insurance giant Allianz signs Claude Code deal with Anthropic | CIO
2026-01-11
Company data provided by crunchbase