Not specified (assumed to be hybrid or flexible based on industry trends).
Tpu networking stack design
Full-stack cross-layer optimization
Ml framework integration jax pytorch
Google is seeking a Software Engineer to join their Tensor Processing Unit (TPU) Host Networking team, focusing on the design and development of networking stacks for machine learning workloads. The ideal candidate should have experience in full-stack development and possess a strong foundation in networking and ML frameworks
Job Summary
This role involves leading the design, development, and deployment of the TPU networking stack from hardware to ML frameworks.
Engineers will perform full-stack cross-layer optimization to enable large-scale training and low-latency inference applications.
The position offers opportunities to switch teams and projects as Google's business grows and evolves.
Matching Summary
Match Score: 85
Google is seeking a Software Engineer to join their Tensor Processing Unit (TPU) Host Networking team, focusing on the design and development of networking stacks for machine learning workloads. The ideal candidate should have experience in full-stack development and possess a strong foundation in networking and ML frameworks.
Salary
Base: $147,000-$211,000; Bonus/Equity: Included; Benefits: Not specified