Software Engineer, TPU Compilers

Company hidden

San Francisco Remote Full-time$380k–$500k / year Posted 2d ago

Above market. This role pays above the $220k median for similar USD roles (799 comparable postings in our corpus).

You’ll have broad technical ownership: identifying performance bottlenecks, choosing the most impactful improvements, and carrying them through implementation and validation. We welcome experience gained through either training or inference, with a strong foundation in optimizing real workloads on TPUs. IN THIS ROLE, YOU WILL: * Optimize model execution on TPUs, improving performance across the complete forward pass and the interactions between model code, kernels, and hardware. * Profile workloads, identify bottlenecks, and implement improvements to latency, throughput, and hardware…

See the company and apply

Unlock to see who’s hiring, read the full description and apply on the company site.

Checking access…

See how well you match this job

Upload your resume and we’ll score your fit for this role — then tailor your CV to it with AI. Free, no credit card.

Check your match

Explore related jobs

Want more roles like this? Browse fresh jobs or tailor your resume with AI.