Senior Applied ML Engineer, Graph Neural Network, ML Frontiers

Google Google · Big Tech · Zürich, Switzerland

Senior Applied ML Engineer focused on Graph Neural Networks within the ML Frontiers team, collaborating on LLM agents and decision forests. The role involves feature development, research exploration, and client collaboration, bridging pioneering models with enterprise solutions.

What you'd actually do

  1. 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).
  2. 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.
  3. Participate in brainstorming and help define future applied research, engineering, and product roadmap priorities.
  4. Focus on Graph Neural Networks within Graph Flow, collaborate with the remainder of the ML Frontiers teamwhile working on LLM agents and decision forests.

Skills

Required

  • software development
  • C++
  • Python
  • machine learning
  • AI

Nice to have

  • data structures
  • algorithms
  • applied machine learning
  • machine learning research
  • PyTorch
  • TensorFlow
  • Jax
  • decision trees

What the JD emphasized

  • 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

Other signals

  • Graph Neural Networks
  • ML Frontiers
  • LLM agents
  • decision forests