Senior Applied ML Engineer, Graph Neural Network, ML Frontiers - Google Careers

Mid

Experience driving progress, solving problems, and mentoring more junior team members; deeper expertise and applied knowledge within relevant area.

• Bachelor's degree or equivalent practical experience.
• 5 years of experience with software development in one or more programming languages.
• 2 years of experience with C++ and Python.
• Experience with machine learning (ML)/AI in a software development environment.

• 5 years of experience with data structures and algorithms.
• 2 years of experience with applied machine learning/machine learning research.
• Experience with one or more neural network frameworks:
  PyTorch, Tensor Flow or Jax.
• Experience with decision trees.

Google Cloud’s mission is to make every business successful through AI by combining cutting-edge technology, infrastructure, and talent. AI/ML software engineers in Cloud bridge the gap between pioneering models and a massive product vehicle reaching billions. Our talent density and AI-powered tools drive rapid development, rooted in a culture of empowerment and a bias to action. In this role, you aren’t just building technology;

you’re shaping the frontier of enterprise and driving the evolution of advanced models.

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.

• Develop feature development (design, coding, doc writing, and maintenance), research exploration (reading papers, benchmarking, and possibly going/submitting to ML conferences), collaborate with clients (from small consultation to large engagements) and partners (e.g., Google Research).
• Implement a distributed graph sampling algorithm (without Flume), benchmark a newly published graph convolution layer in JAX, or develop an interactive Colab component to analyze a trained model.
• Participate in brainstorming and help define future applied research, engineering, and product roadmap priorities.
• Focus on Graph Neural Networks within Graph Flow, collaborate with the remainder of the ML Frontiers team while working on LLM agents and decision forests.

Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents‑to‑be, criminal histories consistent with legal requirements, or any other basis protected by law.

See also Google's EEO Policy , Know your rights: workplace discrimination is illegal , Belonging at Google , and How we hire .

Google is a global company and, in order to facilitate efficient collaboration and communication globally, English proficiency is a requirement for all roles unless stated otherwise in the job posting.

To all recruitment agencies:
Google does not accept agency resumes. Please do not forward resumes to our jobs alias, Google employees, or any other organization location. Google is not responsible for any fees related to unsolicited resumes.