Senior Applied ML Engineer, Graph Neural Network, ML Frontiers

Minimum qualifications:

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

Preferred qualifications:

• 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, TensorFlow or Jax.
• Experience with decision trees.

About the jobGoogle 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.

Responsibilities

• 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 teamwhile working on LLM agents and decision forests.

Google is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. See also Google's EEO Policy and EEO is the Law. If you have a disability or special need that requires accommodation, please let us know by completing our Accommodations for Applicants form .