
About Google
Empowering the world through technology and information
Key Highlights
- Over 100,000 employees globally
- Headquartered in Mountain View, California
- Parent company Alphabet Inc. valued at $1.5 trillion
- Google Cloud Platform serves millions of customers
Google LLC, headquartered in Mountain View, California, is a global leader in internet-related services and products, including its flagship search engine, Google Search, and the Android operating system. With over 100,000 employees, Google also offers cloud computing services through Google Cloud P...
🎁 Benefits
Google offers competitive salaries, equity options, generous PTO policies, comprehensive health benefits, and a remote work policy that allows flexibi...
🌟 Culture
Google is known for its engineering-first culture, emphasizing innovation and collaboration. The company fosters a unique environment that encourages ...
Skills & Technologies
Overview
Google is hiring a Senior Staff Software Engineer for their Torch TPU team to lead technical project strategy and work on machine learning infrastructure. You'll utilize Python and C++ to develop and improve software for ML capabilities.
Job Description
Who you are
You have a Bachelor's degree or equivalent practical experience and at least 8 years of experience in software development. With 7 years of experience leading technical project strategy and working with machine learning design, you are well-versed in ML infrastructure, including model deployment and evaluation. You have a strong background in speech/audio technologies, reinforcement learning, or other specialized ML fields, and you possess 5 years of experience in design, architecture, and testing software products.
Your programming skills in Python and C++ are complemented by your experience with data structures and algorithms, which spans over 8 years. You have also held a technical leadership role for at least 5 years, guiding project teams and setting technical direction. Your experience in performance analysis and debugging, especially for systems that span multiple interconnected hosts, makes you a valuable asset to any team.
You are familiar with contributing to open-source software development, particularly with frameworks like PyTorch or JAX. You thrive in collaborative environments, working alongside partner teams and leading researchers to enhance ML capabilities in Google's offerings.
Desirable
A Master's degree or PhD in Engineering, Computer Science, or a related technical field would be a plus, as would experience in a matrixed organization involving cross-functional projects.
What you'll do
In this role, you will design, develop, test, deploy, maintain, and improve software that supports Google's machine learning initiatives. You will collaborate with various teams to bring ML capabilities to the PyTorch-in-Cloud offering, ensuring that the software you develop can handle information at massive scale. Your responsibilities will include leading technical project strategies, guiding the architecture of ML systems, and contributing to the overall direction of ML infrastructure.
You will also be involved in debugging and fine-tuning models, ensuring that they perform optimally in production environments. Your leadership will help shape the technical direction of projects, and you will mentor junior engineers, fostering a culture of learning and innovation within the team. You will be expected to stay updated with the latest advancements in machine learning and apply them to enhance Google's products and services.
What we offer
Google provides a dynamic work environment where you can make a significant impact on the next generation of technologies. You will have the opportunity to work with cutting-edge tools and collaborate with some of the brightest minds in the industry. The company values innovation and encourages you to bring fresh ideas to the table. You will also benefit from a competitive compensation package and a supportive culture that promotes professional growth and development.
Interested in this role?
Apply now or save it for later. Get alerts for similar jobs at Google.
Similar Jobs You Might Like
Based on your interests and this role

Staff Engineer
Google is hiring a Staff Software Engineer for TPU Performance to analyze and optimize machine learning models on Tensor Processing Units. You'll work with performance analysis, large-scale systems, and debugging techniques. This position requires 8+ years of experience in software development.

Staff Engineer
Google is seeking a Staff Software Engineer for TPU Supercomputer Enablement to provide technical leadership on high-impact projects. You'll work with C++ and large-scale infrastructure. This position requires 8+ years of experience in software engineering.

Software Engineering
Google is hiring a Senior Software Engineer to develop a compiler for Tensor Processing Units designed to accelerate machine learning workloads. You'll work with CUDA C++ and TensorFlow to compile high-performance implementations. This position requires 5+ years of experience in software development.

Staff Engineer
Google is hiring a Staff Software Engineer to work on CPU architecture and software solutions. You'll leverage your expertise in C++, ARM, and x86 to design and optimize large-scale systems. This role requires 8+ years of experience in software engineering and technical leadership.