TPU Kernel Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Anthropic · 23 hours ago

TPU Kernel Engineer

Anthropic is a public benefit corporation focused on creating reliable and beneficial AI systems. The TPU Kernel Engineer will be responsible for identifying performance issues in various ML systems and optimizing kernels for TPUs to enhance system efficiency.

Artificial Intelligence (AI)Foundational AIGenerative AIInformation TechnologyMachine Learning
check
H1B Sponsorednote

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

Qualification

TPU optimizationML systems experienceKernel designComputer architectureML framework internalsPair programmingCommunication skills

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 and benefits
Optional equity donation matching
Generous vacation and parental leave
Flexible working hours

Company

Anthropic

twittertwittertwitter
company-logo
Anthropic is an AI research company that focuses on the safety and alignment of AI systems with human values.

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 Stage
Total Funding
$33.74B
Key Investors
Lightspeed Venture PartnersGoogleAmazon
2025-09-02Series F· $13B
2025-05-16Debt Financing· $2.5B
2025-03-03Series E· $3.5B

Leadership Team

leader-logo
Dario Amodei
CEO & Co-Founder
linkedin
leader-logo
Daniela Amodei
President and co-founder
linkedin
Company data provided by crunchbase