Requirements:

• Bachelor’s degree or equivalent practical experience.
• 5 years of experience with ML infrastructure (e.g., model deployment, model evaluation, optimization, data processing, debugging).
• Experience designing, training, and deploying Graph Neural Networks (GNNs) or graph embedding models using frameworks such as PyTorch Geometric (PyG), JAX/Jraph, or TF-GNN.

Nice to haves:

• Ability to lead the Research and Development (R&D) and production of complex ML/Infra algorithms.

What you'll be doing:

• Identify, develop, and productionize GNN/ML methods driven by client feedback, team brainstorming, literature review, and original research applied into the GraphFlow GNN library.
• Develop and productionize industrial GNN/ML models and infrastructure for the GraphML for Google Systems cross-Product Area initiative to build a unified GNN and AI agent system to support Google's Systems.
• Partner closely with product teams and research groups to translate business requirements into ML solutions.

Perks and benefits:

• Information collected and processed as part of your Google Careers profile, and any job applications you choose to submit, is subject to Google's Applicant and Candidate Privacy Policy.
• Google is proud to be an equal opportunity and affirmative action employer.
• If you have a need that requires accommodation, please let us know by completing our Accommodations for Applicants form.
• Google is a global company; English proficiency is a requirement for all roles unless stated otherwise in the job posting.