Software Engineer, TPU Host Networking

This role is based in Sunnyvale, CA.

Compensation:

$147,000-$211,000 + bonus + equity + benefits

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 various ML workloads.
• Analyze and debug TPU networking performance issues in production.
• Develop and enhance telemetry for network behavior visibility.

Minimum Qualifications:

• Bachelor’s degree or equivalent practical experience.
• 2 years of experience in software development or 1 year with an advanced degree.
• Experience with large-scale infrastructure, distributed systems, or networking.
• 2 years of experience with networking protocols and troubleshooting.
• 2 years of experience working in C++.

Preferred Qualifications:

• Experience with data structures and algorithms.
• Performance optimization experience.
• Experience in network infrastructure and machine learning infrastructure.