Google is a leading technology company that develops next-generation technologies impacting billions of users. In this role, you will collaborate on advanced Machine Learning systems, transforming research expertise into scalable products while optimizing system performance.
Responsibilities
- Collaborate or lead on team projects to carry out design, analysis, and development of advanced ML systems across the stack using your research expertise
- Support building end-to-end ML Systems that involves working across the full stack, from low-level hardware acceleration and compiler optimizations to high-level model architecture and production APIs, transforming your research expertise into robust, scalable products
- Optimize complex system performance by analyzing and fixing performance bottlenecks, memory inefficiencies, and errors in production systems to meet stringent customer goals
- Elevate engineering excellence by writing well-tested code, conducting code reviews and fostering a culture of quality by advocating best engineering practices
Skills
- PhD degree in Computer Science, ML/AI, or a related field, or equivalent practical experience
- Experience coding in one of the following programming languages including but not limited to: Python, C, C++, Java, JavaScript or Golang
- Experience in Machine Learning or Artificial Intelligence
- Research experience in designing, developing, or applying ML/AI systems or applications in a large-scale distributed environment
- Experience in designing, training, or refining complex ML/AI models
- Experience in deep learning frameworks like TensorFlow/Jax/Pytorch
- Experience in building a stack for an AI-powered application, including data ingestion and processing pipelines, building APIs, and connecting the model to a user-facing interface
- Familiarity with model architectures (CNNs, NLP Transformers, Diffusion/Vision Transformers)
- Availability to start full-time role in 2026
Benefits
Company Overview
Google specializes in internet-related services and products, including search, advertising, and software. It is a sub-organization of Alphabet. It was founded in 1998, and is headquartered in Mountain View, California, USA, with a workforce of 10001+ employees. Its website is https://www.google.com.Company H1B Sponsorship
Google has a track record of offering H1B sponsorships, with 8763 in 2025, 8872 in 2024, 9682 in 2023, 11626 in 2022, 9109 in 2021, 9785 in 2020. Please note that this does not guarantee sponsorship for this specific role.
Apply Now