TPU Kernel Engineer

Benefits

Parental leave

Not verified

Non-birth-parent leave

Not verified

Family-building benefits

• Fertility benefits: Not verified
• Adoption assistance: Not verified
• Surrogacy assistance: Not verified

Mental health support

Not verified

Relocation assistance

Not verified

Childcare support

Not verified

Learning budget

Not verified

Verification

Not verified

Salary

Not verified not verified - source not recorded; timestamp not recorded

401(k) match

Not verified

Was this benefit information wrong? Tell us.

Schedule

Shift type

Not verified

Weekend work

Not verified

Application

Cover letter

Not verified

Assessment

Not verified

Deadline

Not stated

Where they hire

State eligibility is not yet verified.

About this role

TPU Kernel Engineer San Francisco, CA | New York City, NY | Seattle, WA About Anthropic Anthropic's mission is to create reliable, interpretable, and steerable AI systems. We want AI to be safe and beneficial for our users and for society as a whole. Our team is a quickly growing group of committed researchers, engineers, policy experts, and business leaders working together to build beneficial AI systems. About the Role As a TPU Kernel Engineer, you'll be responsible for identifying and addressing performance issues across many different ML systems, including research, training, and inference. A significant portion of this work will involve designing and optimizing kernels for the TPU. You will also provide feedback to researchers about how model changes impact performance. Strong candidates will have a track record of solving large-scale systems problems and low-level optimization. You may be a good fit if you: - 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 Strong candidates may also have experience with: - 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

Read the full description at job-boards.greenhouse.io. FewerJobs shows a source-linked preview and links to the original posting.