TPU Kernel Engineer

The shape of it

In the posting’s own words

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.

What it asks for · 6

• 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

Also a plus

• 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 with transformers

Tools and skills named

Words the posting leans on

• performance5×
• systems5×
• kernel4×
• models4×
• accelerators3×
• impact3×
• designing2×
• experience2×
• implement2×
• inference2×
• language2×
• large-scale systems2×
• optimizing2×
• pair2×
• research2×
• significant2×

Counted from the posting after the mission statement and the legal notices are set aside. The ones near the top are the ones a screener is looking for.

More open at Anthropic

• Account Executive, AI NativeNew York City, NY; San Francisco, CA | New York City, NY
• Account Executive - DNBSingapore
• Account Executive, Public SectorSydney, Australia
• Account Executive - Public Sector (ASEAN)Singapore
• Account Executive, StartupsSan Francisco, CA | New York City, NY
• Accounting, Revenue Internal ControlsSan Francisco, CA | Seattle, WA