Job Board
LogoLogo

Get Jobs Tailored to Your Resume

Filtr uses AI to scan 1000+ jobs and finds postings that perfectly matches your resume

Google Logo

Software Engineer - TPU Performance

Google

$141,000 - $202,000
Aug 28, 2025
Sunnyvale, CA, USA
Apply Now

Google is looking to solve the problem of optimizing the performance, power, and energy efficiency of current and future ML workloads, and to design and develop the next-generation ML systems with Hardware and Software co-design and optimization.

Requirements

  • 2 years of experience with software development in one or more programming languages.
  • 2 years of coding experience in one or more of the following languages: C, C++, Java, or Python.
  • 2 years of experience testing, maintaining, or launching software products.
  • 2 years of experience with data structures/algorithms.
  • Experience focused on ML algorithm and performance analysis and optimization.
  • Experience with architecture simulator development and microarchitecture.
  • Knowledge of computer architecture such as TPU's or other accelerators.

Responsibilities

  • Analyze performance, power, and energy efficiency of current and future ML workloads to identify issues.
  • Enable the peak efficiency of future and current ML systems through full-stack ML hardware-software co-design by proposing Hardware-aware algorithm optimization and related simulation modeling.
  • Establish an understanding of the latest business-critical production ML models (e.g., large-language models, large embedding models) to inform optimizations of model architecture, software systems, and hardware architecture.
  • Explore and define future ML accelerator system and chip architectures with objective and data-driven insights.
  • Design, develop, test, deploy, maintain, and enhance software solutions.
  • Manage project priorities, deadlines, and deliverables.
  • Build the Machine Learning (ML) systems with Hardware and Software co-design and optimization.

Other

  • Bachelor’s degree or equivalent practical experience.
  • Excellent communication skills.
  • Display leadership qualities and be enthusiastic to take on new problems across the full-stack.
  • Manage project priorities, deadlines, and deliverables.
  • Work on a specific project critical to Google’s needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve.