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
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
Equity and benefits
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.
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
Recent News
2026-01-17
2026-01-17
Company data provided by crunchbase