TPU Kernel Engineer

Overview

Role involves optimizing ML systems and designing TPU kernels.

Ideal candidate has significant experience with ML systems optimization and low-level performance tuning.

Locations

• United States, New York, New York City
• United States, California, San Francisco
• United States, Washington, Seattle

Requirements

• Bachelor's degree or equivalent experience
• Experience optimizing ML systems for TPUs or GPUs
• Experience with high performance ML systems

Responsibilities

• Identify and address performance issues
• Design and optimize kernels for TPU
• Provide feedback on model performance

Benefits

• Competitive compensation
• Generous vacation
• Parental leave
• Flexible working hours
• Office space for collaboration