Skip to main content
CareerApp

TPU Kernel Engineer

Menlo Ventures Portfolio

New York, NY · full time

$280,000 – $850,000

Listed on Menlo Ventures Portfolio’s own careers site. You apply with them directly — we never stand between you and the employer.

What this role is

This role focuses on optimizing machine learning performance on TPU and other accelerators, designing custom kernels, and solving low-level systems problems for research and production workloads. It suits engineers with deep experience in accelerator optimization and a passion for high-performance computing who want to directly impact large-scale AI infrastructure.

Our summary, not Menlo Ventures Portfolio’s wording. The full posting is on their site.

Skills this role names

Log in to see which of these are already on your profile.

What they ask for

Required

  • Bachelor's degree or equivalent experience
  • Experience optimizing ML systems for accelerators like TPU or GPU
  • Ability to solve large-scale systems problems and low-level optimization

Nice to have

  • High-performance large-scale ML systems experience
  • Kernel design and implementation for ML accelerators
  • Deep understanding of accelerator architecture
  • ML framework internals knowledge
  • Transformer language modeling experience
  • Assembly-level kernel debugging

Turn on analytics and we load Google Analytics: Google gets the pages you open and what you do here — searches, jobs you view, jobs you apply to — and sets its own cookies. Leave it off and the only cookies we set are your login, your theme, and this answer. Accept All also records a yes to advertising, which nothing uses yet. Privacy Policy.