TPU Kernel Engineer

About Anthropic

Anthropic creates reliable, interpretable, and steerable AI systems.

What you'd do

• Identify performance issues in ML systems
• Design and optimize TPU kernels
• Provide feedback on model performance effects
• Debug kernel performance at assembly level

What they're looking for

• Experience with optimizing ML systems for TPUs or GPUs
• Background in computer architecture
• Understanding of ML framework internals

Perks

• Competitive compensation
• Generous vacation and parental leave
• Flexible working hours