TPU Kernel Engineer at Anthropic

Seattle, Washington, United States

Apply Now

$315,000 – $560,000Compensation

Senior (5 to 8 years), Expert & Leadership (9+ years)Experience Level

Full TimeJob Type

UnknownVisa

Artificial Intelligence, Machine LearningIndustries

Requirements

At least a Bachelor's degree in a related field or equivalent experience
Significant experience optimizing ML systems for TPUs, GPUs, or other accelerators
Track record of solving large-scale systems problems and low-level optimization
Results-oriented, with a bias towards flexibility and impact
Willingness to pick up slack, even if it goes outside job description
Enjoy pair programming
Interest in learning more about machine learning research
Care about the societal impacts of work

Responsibilities

Identify and address performance issues across ML systems, including research, training, and inference
Design and optimize kernels for the TPU
Provide feedback to researchers about how model changes impact performance

Skills

Key technologies and capabilities for this role

TPUGPUKernel OptimizationComputer ArchitectureML FrameworksTransformersLow-Precision InferenceCollective CommunicationAssembly DebuggingLLM Sampling

Questions & Answers

Common questions about this position

What is the salary range for the TPU Kernel Engineer position?

The expected salary range is $315,000 - $560,000 USD annually.

What is the location and hybrid work policy?

The role follows a location-based hybrid policy where staff are expected to be in one of the offices at least 25% of the time, though some roles may require more.

What skills and experience are required for this role?

Candidates need significant experience optimizing ML systems for TPUs, GPUs, or other accelerators, a track record of solving large-scale systems problems and low-level optimization, plus traits like being results-oriented, flexible, and enjoying pair programming.

What is the company culture like at Anthropic?

Anthropic has a quickly growing team of researchers, engineers, policy experts, and business leaders focused on safe and beneficial AI, with a culture that values pair programming, flexibility, picking up slack beyond job descriptions, and caring about societal impacts.

What makes a strong candidate for this position?

Strong candidates have a track record in large-scale systems and low-level optimization, experience with TPU kernel design, and are encouraged to apply even if they don't meet every qualification, especially those from underrepresented groups.

Anthropic

Develops reliable and interpretable AI systems

About Anthropic

Anthropic focuses on creating reliable and interpretable AI systems. Its main product, Claude, serves as an AI assistant that can manage tasks for clients across various industries. Claude utilizes advanced techniques in natural language processing, reinforcement learning, and code generation to perform its functions effectively. What sets Anthropic apart from its competitors is its emphasis on making AI systems that are not only powerful but also understandable and controllable by users. The company's goal is to enhance operational efficiency and improve decision-making for its clients through the deployment and licensing of its AI technologies.

San Francisco, CaliforniaHeadquarters

2021Year Founded

$11,482.1MTotal Funding

GROWTH_EQUITY_VCCompany Stage

Enterprise Software, AI & Machine LearningIndustries

1,001-5,000Employees

Benefits

Flexible Work Hours

Paid Vacation

Parental Leave

Hybrid Work Options

Company Equity

Risks

Ongoing lawsuit with Concord Music Group could lead to financial liabilities.

Technological lag behind competitors like OpenAI may impact market position.

Reliance on substantial funding rounds may indicate financial instability.

Differentiation

Anthropic focuses on AI safety, contrasting with competitors' commercial priorities.

Claude, Anthropic's AI assistant, is designed for tasks of any scale.

Partnerships with tech giants like Panasonic and Amazon enhance Anthropic's strategic positioning.

Upsides

Anthropic's $60 billion valuation reflects strong investor confidence and growth potential.

Collaborations like the Umi app with Panasonic tap into the growing wellness AI market.

Focus on AI safety aligns with increasing industry emphasis on ethical AI development.

Land your dream remote job 3x faster with AI

Try Jobo Free