Requirements

• 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++.

Nice to haves

• 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.

What you'll be doing

• Work on AI framework development to 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 to enable training and deployment of advanced machine learning models.
• Collaborate with Google teams and leading researchers to deliver cutting-edge ML capabilities to the PyTorch-in-Cloud offering.
• Help with design, development, testing, deploying, and maintaining software.