Google Cloud accelerates every organization’s ability to digitally transform its business and industry. We deliver enterprise-grade solutions that leverage Google’s cutting-edge technology, and tools that help developers build more sustainably. Customers in more than 200 countries and territories turn to Google Cloud as their trusted partner to enable growth and solve their most critical business problems.
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.
Collaborate with other Google teams and leading researchers across the industry to continuously bring cutting-edge ML capabilities to our PyTorch-in-Cloud offering.
Help with design, development, testing deploying and maintaining software.
Minimum qualifications
Bachelor’s degree or equivalent practical experience.
2 years of experience with software development in one or more programming languages, or 1 year of experience with an advanced degree.
1 year of experience with ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging).
Experience in C++.
Preferred qualifications
Master's degree or PhD in Computer Science or related technical fields.
Experience with ML frameworks, such as PyTorch/XLA/Pallas/GPU/TPU programming.
Experience with compilers, machine learning optimization, distributed machine learning.
Bachelor’s degree or equivalent practical experience.
2 years of experience with software development in one or more programming languages, or 1 year of experience with an advanced degree.
1 year of experience with ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging).
Experience in C++.
