Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Staff Software Engineer, TPU Machine Learning Supercomputer

$207k - $301k

Google

Design, develop, test, deploy, and debug critical system software that enables TPU Machine Learning accelerators to function seamlessly.Develop advanced analytics and health management capabilities to effectively manage and optimize large-scale ML systems.Lead high-impact projects and steer successful delivery while ensuring alignment with broader team strategies.Provide technical guidance and mentorship to software engineers to foster their professional growth and development.Minimum qualifications:Bachelor's degree or equivalent practical experience.8 years of experience with software development in C++ or Go.5 years of experience with large-scale infrastructure, distributed systems, or networks, as well as testing and launching software products.3 years of experience with software design and architecture.Experience with operating systems, data structures, and algorithms.Experience developing, integrating, and testing system and user-space software (including tools, dashboards, and monitoring) for hardware accelerators or TPU systems.Preferred qualifications:Master’s degree or PhD in Engineering, Computer Science, or a related technical field.8 years of experience with data structures and algorithms.3 years of experience in a technical leadership role leading project teams and setting technical direction.3 years of experience working in a complex, matrixed organization involving cross-functional, or cross-business projects.Experience building backend software for high-performance computing (HPC) and machine learning (ML) applications, including knowledge of data analytics, ML architecture, and how common algorithms map to software/hardware operations.Understanding of highly distributed systems, control plane and management Software, and networking concepts.Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond web search. We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google’s needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to be versatile, display leadership qualities and be enthusiastic to take on new problems across the full-stack as we continue to push technology forward. As a member of the TPU Machine Learning Supercomputer (MLSC) team, you will design and develop features to significantly improve the scalability and reliability of large-scale software across TPUs and other distributed networked hardware machines. Your work will span various layers of the software stack, from host daemons to network routing and distributed control software running across Google's internal and cloud infrastructure. You will also provide leadership to help formulate and drive software development plans for future supercomputer generations.The AI and Infrastructure team is redefining what’s possible. We empower Google customers with breakthrough capabilities and insights by delivering AI and Infrastructure at unparalleled scale, efficiency, reliability and velocity. Our customers include Googlers, Google Cloud customers, and billions of Google users worldwide.We're behind Google's groundbreaking innovations, empowering the development of AI models, delivering unparalleled computing power to global services, and providing the essential platforms that enable developers to build the future. From software to hardware our teams are shaping the future of world-leading hyperscale computing, with key teams working on the development of our TPUs, Vertex AI for Google Cloud, Google Global Networking, Data Center operations, systems research, and much more.Individual pay is determined by factors including job-related skills, experience, and relevant education or training. US: $207000 - $301000 (USD) + 20% bonus target + equity + benefitsLearn more about benefits at Google.Bachelor's degree or equivalent practical experience.8 years of experience with software development in C++ or Go.5 years of experience with large-scale infrastructure, distributed systems, or networks, as well as testing and launching software products.3 years of experience with software design and architecture.Experience with operating systems, data structures, and algorithms.Experience developing, integrating, and testing system and user-space software (including tools, dashboards, and monitoring) for hardware accelerators or TPU systems.

Vacancy posted 2 days ago
Similar jobs that could be interesting for youBased on the Staff Software Engineer, TPU Machine Learning Supercomputer in Sunnyvale, CA vacancy
  • $207k - $300k

     ...guide sustainable engineering choices for ML systems...  ...of experience in software development.5...  ...voice), reinforcement learning (e.g., sequential...  ...with TPUs, TPU system design, and...  ...software solutions.As a Staff Software Engineer,...  ...organization, the Machine Learning Infrastructure... 
    Suggested

    Google

    Mountain View, CA
    4 days ago
  • $207k - $300k

    Lead the design and end-to-end software delivery of specialized AI compute platforms, ensuring high availability for massive-scale workloads...  ...distributed hardware platforms.Partner closely with hardware engineering and chip design teams to influence future architectures for... 
    Suggested
    Remote work
    Worldwide

    Google

    Sunnyvale, CA
    1 day ago
  • $207k - $301k

     ...Programming Interfaces (APIs), Software Development Kits (SDKs), and...  ...of a generalist Software Engineers, specifically around optimizing...  ...human voice), reinforcement learning (e.g., sequential decision making...  ...(e.g., GPU / Pixel TPU / NPUs / CPU) on Android, Chrome... 
    Suggested
    Shift work

    Google

    Sunnyvale, CA
    6 hours ago
  • $262k - $365k

     ...various accelerators (GPU /Pixel TPU /NPUs/CPU) on Android,...  ...experience.8 years of experience in software development.7 years of...  ...products.Experience with machine learning infrastructure, C++, performance...  ...:Master’s degree or PhD in Engineering, Computer Science, or a... 
    Suggested

    Google

    Sunnyvale, CA
    3 days ago
  • $192k - $279k

     ...optimization of the hardware/software stack to enable...  ...in Electrical Engineering, Computer...  ...Experience with deep learning frameworks such as...  ...opportunity to drive TPU (Tensor Processing...  ...systems.As a Staff Co-Design Engineer...  ...roadmap for our future machine learning serving... 
    Suggested
    Worldwide

    Google

    Sunnyvale, CA
    3 days ago
  • $138k - $197k

    Define TPU microarchitecture and develop high-quality, performant...  ...'s degree in Electrical Engineering, Computer Engineering,...  ...accelerators powering our AI and machine learning workloads in data centers. You...  ...to build the future. From software to hardware our teams are shaping... 
    Worldwide

    Google

    Sunnyvale, CA
    1 day ago
  • $207k - $300k

     ...Influence and coach a distributed team of engineers.Facilitate alignment and clarity...  ..., maintain, and enhance large-scale software solutions.Minimum qualifications:...  ...silicon validation for inhouse SoCs, machine learning (ML) Systems (TPU/GPU) and compute servers. You will get... 
    Worldwide

    Google

    Sunnyvale, CA
    1 day ago
  • $207k - $301k

     ...driving continuous improvements to the machine learning software/hardware stacks through providing...  ...such as horizontal scaling for upcoming TPU chips) by making contributions across...  ...qualifications:Master’s degree or PhD in Engineering, Computer Science, or a related technical... 
    Worldwide

    Google

    Sunnyvale, CA
    6 hours ago
  • $207k - $301k

    Lead the end-to-end development of novel machine learning models, incorporating techniques like...  ...practical experience.8 years of experience in software development.5 years of experience...  ...recommender systems.Google's software engineers develop the next-generation technologies... 

    Google

    Mountain View, CA
    6 hours ago
  • $200k - $250k

     ...leveraging its commercial self-driving software to develop, test and deploy...  ...Department of Defense.Kodiak is recruiting a Staff Software Engineer to join our Planning team and...  ...organizationIntegrate engineered driving logic and machine-learning modelsDevelop robust models to... 
    Temporary work
    Work at office
    Visa sponsorship

    Kodiak Robotics

    Mountain View, CA
    4 days ago
  • $207k - $301k

     ...Influence and coach a distributed team of engineers.Facilitate alignment and clarity across...  ..., maintain, and enhance large scale software solutions.Minimum qualifications:Bachelor...  ..., Angular, Java, Spanner, and Machine Learning. You will guide system design, architecture... 

    Google

    Sunnyvale, CA
    1 day ago
  • $207k - $301k

    Design new deep learning based ML models to predict users’ interaction...  ...channels.Incorporate and engineer input features from ad creatives...  ....8 years of experience in software development.5 years of experience...  ...growth. We focus on building machine learning models to predict... 

    Google

    Mountain View, CA
    1 day ago
  • $156k - $255k

     ...infrastructure domain, aligning engineering efforts with medium to long...  ...requirements. You will learn to navigate complex organizational...  ...You will produce high quality software that is unit tested, code...  ...storage systems, or advanced machine learning infrastructure. Proven... 
    For contractors
    Work experience placement
    Work at office
    Flexible hours

    Linkedin

    Mountain View, CA
    3 days ago
  • $207k - $301k

     ...experience. 8 years of experience in software development.5 years of...  ...:Master’s degree or PhD in Engineering, Computer Science, or a...  ...experience with LLM evaluation, machine learning algorithms and tools, and...  ...evaluation technology to Cloud. As a Staff Software Engineer on this... 

    Google

    Sunnyvale, CA
    3 days ago
  • $157.1k - $246.5k

     ...with one of our offices if preferred. Role Overview:As a Staff Software Engineer on the Simulation Test Engineering team, you will architect...  ...libraries. Experience with computational geometry, machine learning, or advanced analytics. “Company Vehicle: Upon successful... 
    Full time
    Local area
    Remote work
    Work from home
    Flexible hours

    General Motors

    Sunnyvale, CA
    1 day ago
  • $152k

     ...connected world. Role Overview The Resource Fabric Engineering team builds and operates the foundational...  ...tooling, and observability solutions that support machine learning innovation at scale. We are looking for a Staff Software Engineer to lead the design and evolution of... 
    Temporary work
    Flexible hours

    Coupang

    Mountain View, CA
    6 hours ago
  • $143k - $286k

     ...WalmartBusiness Segment: Home OfficePosition Summary...The Staff, Software Engineer plays a critical role in delivering scalable platform...  ...platform capabilities including infrastructure, tooling, and machine learning frameworks aligned with roadmap priorities.Drive full... 
    Full time
    Temporary work
    Part time

    Walmart

    Sunnyvale, CA
    6 hours ago
  • $185k - $284.1k

     ...analyses to evaluate autonomous driving software performance across the autonomy stack....  ...efforts with autonomy, systems engineering, simulation, and data teams to embed evaluation...  ...Mechanical or Aerospace Engineering, Machine Learning, or a related field. What Will Give... 
    Full time
    Local area
    Remote work
    Work from home
    Relocation
    Relocation package
    Flexible hours

    General Motors

    Sunnyvale, CA
    1 day ago
  • $207k - $300k

    Write low-level performance software for a new systolic array accelerators...  ...that are a match to this new machines. Use that direct experience...  ...for the performance software engineers and general software...  ...systems, analytics, machine learning, working with distributed query... 

    Google

    Mountain View, CA
    6 hours ago
  • $176.4k - $319.72k

     ...Price, and other leading investors.About the RoleAs a Senior/Staff Software Engineer working on driving behavior verification, you are...  ...behavior of an autonomous vehicle.Leverage a combination of machine learning (ML) models and safety metrics from literature to evaluate... 
    Odd job
    Work experience placement
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    2 days ago
  • $207k - $301k

     ...the design and development of large-scale software services that revolutionize data analytics, with a focus on Machine Learning and Generative AI applications.Collaborate...  ...Preferred qualifications:Master’s degree or PhD in Engineering, Computer Science, or a related technical... 

    Google

    Sunnyvale, CA
    6 hours ago
  • $189k - $303k

     ...efficient and accessible for all. We’re searching for a Staff Machine Learning Engineer - Online Maps to join Aurora’s Maps ML team. The Maps ML...  ...powers the Aurora DriverRequired QualificationsExcellent software engineering skills in Python and/or C++Extensive exp in any... 
    Work at office
    Local area
    3 days per week

    Aurora Innovation

    Mountain View, CA
    2 days ago
  • $207k - $301k

     ...practical experience.8 years of experience in software development.8 years of experience with...  ...tuning).Experience with multimodal machine learning or LLMs, including building and...  ...quality best practices.Google's software engineers develop the next-generation technologies... 

    Google

    Mountain View, CA
    6 hours ago
  • $240k - $265k

     ...leveraging its commercial self-driving software to develop, test and deploy...  ...of Defense.We are looking for a Staff Software Engineer to help shape how learned models are integrated into behavior...  ...the intersection of Planning and Machine Learning, working closely with ML... 
    Visa sponsorship

    Kodiak Robotics

    Mountain View, CA
    3 days ago
  • $190k - $300k

     ...technology. We are at the forefront of software and hardware innovation, pushing the...  .... Will consider remote.The Role: Staff Software Engineer, SIMD KernelsWhat you will do:As a member...  ...structures, system software, and machine learning fundamentals.Proficient in C/C++ and... 
    Work experience placement
    Remote work

    d-Matrix

    Santa Clara, CA
    2 days ago
  • $190.9k - $334.1k

     ...DescriptionIt all started when engineer Fred Luddy wrote code that...  ...are looking for a software engineer and technical lead...  ...experience as a technical lead, Staff Engineer, or equivalent senior...  ...platformsKnowledge of MLOps or applying machine learning models to production use... 
    Work at office
    Immediate start
    Remote work
    Flexible hours

    ServiceNow

    Santa Clara, CA
    3 days ago
  •  ...is powering the next generation of supercomputing, high-performance computing, cloud,...  ...forward.THE ROLEAMD is seeking a Senior Staff Software Engineer to lead compiler development and...  ..., tensor or graph compilers, machine learning frameworks, or high-performance computing... 

    AMD

    San Jose, CA
    1 day ago
  • $207k - $340k

     ...the business needs of the team.LinkedIn's AI and Machine Learning Engineers are both data/research scientists and software engineers, who develop and implement machine...  ...from ideation to implementation. As a Principal Staff Software Engineer you will be setting the technical... 
    For contractors
    Work experience placement
    Work at office
    Flexible hours

    Linkedin

    Mountain View, CA
    1 day ago
  • $193.93k - $352.29k

     ...and other leading investors.About the RoleOur software team is growing, and we are looking for talented engineers to join us and be instrumental to one of the following...  ...systems; modern microservice architecture; machine learning, multithreading; x86 architecture; and... 
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    2 days ago
  • $193.93k - $352.29k

     ...deploy core infrastructure components in machine learning model life cycle, to push the...  ...model compression. Work with autonomy engineers to optimize, validate, and deploy large...  ...framework, FTL.Write robust, high quality software to increase our confidence in our vehicle... 
    Work experience placement
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    6 hours ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff Software Engineer, TPU Machine Learning Supercomputer. Be the first to apply!