Google

Software Engineer, TPU Host Networking

Google$147K — $211K *
Information Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • Bachelor's degree or equivalent practical experience.
  • 2 years of experience in software development or 1 year with an advanced degree in a related field.
  • 2 years of experience with large-scale infrastructure or distributed systems.
  • 2 years of experience with networking protocols and troubleshooting.
  • 2 years of experience programming in C.

Responsibilities

  • Write code for product or system development.
  • Design, develop, test, and deploy the TPU networking stack.
  • Optimize TPU networking performance for various ML workloads.
  • Analyze and debug TPU networking performance issues in production.
  • Enhance telemetry for improved visibility into network behavior.

Benefits

  • Access to cutting-edge ML frameworks like JAX and PyTorch.
  • Opportunity to work on innovative TPU technologies.
  • Involvement in large-scale training and low-latency applications.
  • Potential for stock options as part of the compensation package.
  • Comprehensive benefits package including health and wellness programs.
Full Job Description
Minimum qualifications:
  • Bachelor's degree or equivalent practical experience.
  • 2 years of experience with software development or 1 year of experience with an advanced degree in an industry setting.
  • 2 years of experience with developing large-scale infrastructure, distributed systems or networks, or experience with compute technologies, storage or hardware architecture.
  • 2 years of experience with networking protocols and troubleshooting.
  • 2 years of experience working in C .

Preferred qualifications:
  • 2 years of experience with data structures and algorithms.
  • 2 years of experience with performance optimization.
  • 2 years of experience with networking protocols.
  • Experience in network infrastructure.
  • Experience with machine learning infrastructure.


About the job

As a team member in Tensor Processing Unit Host Networking, you will play a leading role in the design, development, testing, deployment, and debugging of the TPU networking stack, from hardware (Tensor Processing Unit, Network Interface Controller) all the way up to ML frameworks (JAX, PyTorch) to enable both large-scale training and low-latency inference applications.

Individual pay is determined by factors including job-related skills, experience, and relevant education or training.

US: $147000 - $211000 (USD) 15% bonus target equity benefits

Learn more about benefits at Google .

Responsibilities
  • Write product or system development code.
  • Design, develop, test and deploy TPU networking stack.
  • Perform full-stack cross-layer optimization of TPU networking performance for a variety of ML workloads.
  • Analyze and debug TPU networking performance issues in production.
  • Develop and enhance telemetry to provide deep visibility into network behavior and accelerate troubleshooting.


About Google

Google is a multinational technology company that specializes in Internet-related services and products. These include online advertising technologies, search engine, cloud computing, software, and hardware. Google was founded in 1998 by Larry Page and Sergey Brin while they were Ph.D. students at Stanford University. The company has grown tremendously since then and has become one of the most valuable companies in the world. Google's mission is to organize the world's information and make it universally accessible and useful.
Learn more about Google
Size
156,500 employees
Market Cap
$1,115.4 billion
Industry
Net Income
$40.2 billion
Founded
1998
5 Year Trend
+23.3%
Revenue
$182.5 billion
NASDAQ

Similar Jobs

More Jobs at Google

More Information Technology Jobs

Find similar Software Engineer, TPU Host Networking jobs: