Senior Software Engineer, TorchTPU

Requirements

• Bachelor's degree in Computer Science or related technical field, or equivalent practical experience.
• 5 years of experience in software development.
• Experience programming in C++.
• Experience in leading cross-functional software oriented projects from inception to delivery.

Nice to haves

• 5 years of experience with data structures and algorithms.
• Experience with ML frameworks (e.g., TensorFlow, JAX, PyTorch), and hardware performance analysis with TPUs and GPUs.
• Experience programming in Python.
• Experience with compilers and compiler construction.
• Ability to effectively navigate and communicate in a fast-moving environment.

What you'll be doing

• Work on AI framework development to successfully enable PyTorch models to run on Google Cloud's TPUs and GPUs and tune for peak performance.
• Provide comprehensive support for ML frameworks and compilers on Cloud TPUs and GPUs, enabling the training and deployment of the most advanced machine learning models, driving innovation and breakthroughs.
• Enable PyTorch models at massive scale for generative models, computer vision (image recognition, object detection, image generation), machine translation, language modeling, rankings and recommendations, speech recognition, etc.
• Collaborate with other Google teams and leading researchers across the industry to continuously bring ML capabilities to the PyTorch-in-Cloud offering.
• Contribute to open-source software development.

Perks and benefits

• Information collected and processed as part of your Google Careers profile, and any job applications you choose to submit are subject to Google's Applicant and Candidate Privacy Policy.
• Google is proud to be an equal opportunity and affirmative action employer.
• If you have a need that requires accommodation, please let us know by completing our Accommodations for Applicants form.
• Google is a global company and, in order to facilitate efficient collaboration and communication globally, English proficiency is a requirement for all roles unless stated otherwise in the job posting.
• To all recruitment agencies: Google does not accept agency resumes.