TPU Kernel Architect

Anthropic; Seattle, New York, San Francisco (WA, NY, CA); On-site; USD 315,000 - 560,000; Full time

Benefits offered by this job

Competitive compensation

Generous vacation and parental leave

Flexible working hours

Job summary

Join a leading company in AI research as a TPU Kernel Engineer focused on optimizing performance across machine learning systems. This position is integral in designing and optimizing TPU kernels, providing feedback to improve model performance, and requires expertise in large-scale systems. With competitive compensation and benefits, this role offers a unique opportunity to contribute to groundbreaking advancements in AI.

Qualifications

• Significant experience optimizing ML systems for TPUs, GPUs, or other accelerators.
• Strong track record in solving large-scale systems problems.
• Experience with high performance, large-scale ML systems is beneficial.

Responsibilities

• Identify and address performance issues across various ML systems.
• Design and optimize kernels for the TPU.
• Collaborate with researchers on model performance.

Skills

Optimizing ML systems for TPUs

Performance tuning

Problem-solving

Education

Bachelor's degree in a related field

Job description

Join a leading company in AI research as a TPU Kernel Engineer focused on optimizing performance across machine learning systems. This position is integral in designing and optimizing TPU kernels, providing feedback to improve model performance, and requires expertise in large-scale systems. With competitive compensation and benefits, this role offers a unique opportunity to contribute to groundbreaking advancements in AI.