- 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, TensorFlow 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.
In this role, you will help define new features based on client interactions and your own research. When a new feature or idea is identified, you will take technical leadership of it. This includes validating the idea through literature reviews and experiments, implementing it efficiently for target hardware (CPU, GPU, TPU, or distributed systems), and productionizing it into Graph Flow as an easy-to-use, modular component.
Discovering new features and ideas requires constant exploration, brainstorming with our team and research collaborators, and gathering client feedback. You will also mentor and help grow other engineers, research students, and interns.
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 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.
