The Core ML team at Google is looking to improve the PyTorch ML framework, processes, ecosystem, and model performance to provide large-scale, cloud-based access to Google’s first-party ML supercomputers (TPUs and TPU Pods) for training and inference workloads.
Requirements
- 5 years of experience with ML design and ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning)
- 5 years of experience in software development
- 5 years of experience testing, and launching software products, and 3 years of experience with software design and architecture
- 8 years of experience with data structures/algorithms
- Experience with compilers or ML frameworks
- Experience with PyTorch and JAX
Responsibilities
- 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 Graphics Processing Units (GPUs), enabling the training and deployment of the most advanced machine learning models, managing innovation and breakthroughs.
- Enable PyTorch models 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 our PyTorch in Cloud offering.
- Design, develop, test, deploy, maintain, and improve software while contributing to open-source software development.
Other
- Bachelor’s degree or equivalent practical experience
- Master’s degree or PhD in Engineering, Computer Science, or a related technical field
- 3 years of experience in a technical leadership role leading project teams and setting technical direction
- 3 years of experience working in an organization involving cross-functional, or cross-business projects
- Ability to work in a team and collaborate with other Google teams and leading researchers across the industry
