The Software Engineer III position at YouTube focuses on AI/ML and TPU efficiency. The role involves optimizing models that power the YouTube algorithm, enhancing model efficiency, and collaborating with ML model developers. Engineers will work on algorithmic efficiency techniques and engage in model and TPU compiler co-design. This position requires a blend of technical expertise and leadership qualities to drive innovation in machine learning infrastructure.

• Bachelor's degree or equivalent practical experience.
• 2 years of experience with ML infrastructure, including model deployment and evaluation.
• 2 years of experience with accelerator hardware architectures such as TPUs or GPUs.
• 2 years of experience programming in C++ and Python.
• Experience with ML frameworks such as TensorFlow, JAX, and PyTorch.

• Analyze ML workloads to identify compute and memory bandwidth bottlenecks.
• Optimize accelerator utilization to maximize compute efficiency.
• Implement and productionize algorithmic efficiency techniques.
• Optimize auxiliary serving and distributed data pipelines.
• Collaborate with ML model developers to co-design hardware-friendly model architectures.

• Employees at Google are often offered benefits like comprehensive health insurance, 401(k) matching, and flexible work arrangements, among other benefits.