2 days ago
London, United KingdomSenior
Responsibilities
- Develop AI framework capabilities that enable PyTorch models to run on Google Cloud TPUs and GPUs and tune them for peak performance.
- Support ML frameworks and compilers on Cloud TPUs and GPUs for machine learning model training and deployment.
- Collaborate with Google teams and industry researchers to advance PyTorch cloud capabilities.
- Design, develop, test, deploy, and maintain software.
Requirements
- Bachelor’s degree or equivalent practical experience.
- At least 2 years of software development experience in one or more programming languages, or 1 year of experience with an advanced degree.
- At least 1 year of experience with ML infrastructure, such as model deployment, model evaluation, data processing, or debugging.
- Experience in C++.
- Preferred: master’s degree or PhD in Computer Science or a related technical field.
- Preferred: experience with ML frameworks such as PyTorch, XLA, Pallas, GPU, or TPU programming.
- Preferred: experience with compilers, machine learning optimization, or distributed machine learning.
Categories
About Google
Google builds consumer and enterprise software and services including Search, Android, YouTube, Chrome, Maps, Gmail, and Google Cloud. Its business model centers on digital advertising and paid cloud, software, and hardware offerings (e.g., Pixel and Nest) for consumers, developers, and organizations. Founded in 1998 and headquartered in Mountain View, California, Google operates globally as a subsidiary of Alphabet Inc.