Google logo

Software Engineer, ML Compilers, Silicon Engineering

Google · Posted Oct 13

Search engine, cloud computing, software, hardware, and artificial intelligence services

Hyderabad, Telangana, IndiaFull-timeOnsiteSenior Level2+ years₹40.0L–₹75.0L yearly41 applicants
InternetCloud ComputingSoftwareHardwareArtificial IntelligenceAdvertisingPublic Company
Be an early applicantFull time

Get a personal compatibility score

Add a resume for personal matches

About the role

The Platforms and Devices team encompasses Google's computing software platforms across desktop, mobile and applications, as well as first-party devices and services combining Google AI, software and hardware. This role focuses on developing ML compilers for the Tensor TPU to accelerate Generative AI and other complex machine learning models running on custom hardware accelerators. The engineer will build compiler tooling that maps ML models to the hardware ISA, root-cause correctness and performance issues on EdgeTPU, and work with model owners to optimize model architectures.

What you will do

  • Build compilers and tools that efficiently map ML models (with a particular focus on always-on computing use cases) to the hardware ISA.
  • Propose and implement fixes to the compiler to address these issues in collaboration with other compiler engineers.
  • Triage and root-cause correctness and performance issues encountered while enabling ML models on EdgeTPU.
  • Interact closely with model owners to influence their model architectures to make them run efficiently on the EdgeTPU.

Skills used in this role

ML CompilersTensor TPUEdge TPUCompiler OptimizationPerformance OptimizationData StructuresAlgorithmsHardware ISAMachine LearningGenerative AI

Preferred qualifications

  • Master's degree or PhD in Computer Science or related technical fields.
  • 2 years of experience with data structures and algorithms.
  • Experience in optimizing ML model inference on device.
  • Experience in power and performance optimizations.
  • Good understanding of hardware, especially hardware that provides a high degree of parallelism.

Benefits and support

  • Compensation and benefits are detailed in the job posting

About Google

Google LLC is an American multinational technology company focusing on artificial intelligence, online advertising, search engine technology, cloud computing, computer software, quantum computing, e-commerce, and consumer electronics. As a subsidiary of Alphabet Inc., it is widely recognized for services like Google Search, YouTube, Android, and the Gemini AI ecosystem.

Industry
Internet
Company size
180000+ employees
Founded
1998-09-04
Location
Mountain View, California, USA
Funding stage
Public Company

Funding

Public Company · $25M raised

Kleiner PerkinsSequoia Capital
  • 1999-06-07Series A$25M

Leadership

SP
Sundar Pichai

Chief Executive Officer

AA
Anat Ashkenazi

Senior Vice President and Chief Financial Officer

PS
Philipp Schindler

Senior Vice President and Chief Business Officer

TK
Thomas Kurian

Chief Executive Officer, Google Cloud