Back to jobs
Job Description
- Lead and manage a team of software engineers, promoting a collaborative culture and psychological safety.
- Coach and mentor engineers to achieve their potential while aligning team execution with TorchTPU priorities and organizational goals.
- Collaborate with global peer managers and teams to drive AI framework development, enabling PyTorch models to run with peak performance on Cloud TPUs.
- Deliver end-to-end performance compiler optimizations and contribute to open-source software, supporting advanced ML frameworks and compilers on Cloud TPUs and GPUs.
- Enable PyTorch models at massive scale for generative models, computer vision, language modeling, and other advanced machine learning applications.