Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another…

…REQUIRED QUALIFICATIONS Core Engineering Strength: Strong production-level coding skills in Python and deep learning frameworks (like PyTorch or JAX). ML & LLM Expertise: Hands…

…focus on TPU architecture and its integration within AI/ML-driven systems. As a Senior Staff Co-Design Engineer on the TPU Chip Architecture…

…You will be instrumental in creating the essential tooling that enables our ML engineers and customers to understand workload behavior, identify performance bottlenecks, and…

…Master’s degree or PhD in Engineering, Computer Science, or a related technical field. Experience with on-device ML Software Development Kits (SDKs)/tooling…

…compiler optimization passes. Experience with debugging correctness and performance issues at all levels of the ML Software (SW) stack. Experience with GPU or TPU…

…software development best practices. We prefer: Knowledge of TPU and GPU architectures and how to optimize code for them. Familiarity with ML compilers like…

…Neuron, TPU or other AI acceleration hardware - Experience directly managing scientists or machine learning engineers - Experience debugging, profiling, and implementing best software engineering practices…

…software engineering experience, with significant time as the technical lead or anchor on a platform, inference runtime, or ML infrastructure team Experience with ML…

…compilers, or formal methods Experience with PyTorch and large-scale distributed training; performance profiling and optimization of ML systems CUDA / GPU or TPU kernel…

…This is all enabled by edge software stack, the AWS Neuron Software Development Kit (SDK), which includes an ML compiler, runtime and natively integrates…

…programming with strong C++ proficiency - Embedded software engineering experience or work on resource-constrained systems - Understanding of ML fundamentals at the linear algebra level…

…users. - Work closely with hardware architects, firmware engineers, driver developers, compiler engineers, and ML application engineers to understand their needs, define tool requirements, and…