Anthropic · 18 hours ago
TPU Kernel Engineer
Anthropic is a public benefit corporation focused on creating reliable and beneficial AI systems. As a TPU Kernel Engineer, you will identify and address performance issues in various ML systems, design and optimize kernels for TPUs, and provide feedback to researchers on model performance.
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
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
Optional equity donation matching
Generous vacation and parental leave
Flexible working hours
A lovely office space in which to collaborate with colleagues
Company
Anthropic
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 StageTotal Funding
$33.74BKey Investors
Fidelity,ICONIQ Capital,Lightspeed Venture PartnersLightspeed Venture PartnersGoogle
2025-09-02Series F· $13B
2025-05-16Debt Financing· $2.5B
2025-03-03Series E· $3.5B
Recent News
2026-02-07
2026-02-07
TechCrunch
2026-02-07
Company data provided by crunchbase