Head of ML/AI Exploration, Architecture, & Engineering

Software EngineeringMountain View, CA (HQ)

About the company

X is Alphabet’s moonshot factory where uncomfortably ambitious, world-changing ideas are developed and taken out into the world. X projects have the potential to transform the lives of billions of people and make the world a radically better place. For X to be successful, we have to repeatedly turn breakthrough technology ideas into products that solve large problems in the world, and create meaningful businesses for Alphabet.

About the team

X’s central Applied AI team is looking for a leader motivated by a desire to make a difference in the world. You will apply your expertise to diverse X projects, tackling domain-specific problems with cutting-edge AI.

About the role:

As the Head of the Applied AI team, your work is to discover, invent, solve, and prototype. From devising experiments and building robust implementations to designing new algorithms, You’ll lead a small team of Research Scientists and work on applying artificial intelligence to critically important, often ambiguous, problems. Our team values individuals who can bridge the gap between innovative research and practical, impactful solutions.

How you will make 10x impact:

  • Attract and develop a diverse team of ML/AI experts, as well as guide and coach them on all aspects of their research. Foster a collaborative environment for brainstorming, advising, and knowledge sharing.
  • Partner on the technical vision and strategy for our different early stage and early investigation projects; co-design and co-lead a technical roadmap.
  • Develop, implement, and deploy robust, scalable ML/AI solutions for real-world problems across multiple X projects and scientific domains, utilizing state of the art developments and techniques with a focus on foundation models and transformer architectures.
  • Design and build hybrid AI models, potentially integrating LLMs with other tools and techniques (e.g., graph-based ML, reinforcement learning).
  • Provide research and engineering leadership on X early explorations: defining problem scope, identifying optimal AI/ML approaches, and evaluating their merits.
  • Guide data strategy and contribute to ML pipeline development and system evaluation.
  • Contribute to code reviews, establish best practices for ML engineering, and optimize code for performance and scalability.
  • Mentor and guide other team members, act as X’s link to the broader research and academic community, as well as to Alphabet’s other research groups, staying abreast of and integrating the latest advancements.

What you should have:

  • PhD in Computer Science, Artificial Intelligence, or a related technical field.
  • Experience leading a team of research scientists and in mentoring and guiding junior researchers or engineers.
  • Experience in applying ML/AI techniques to complex problems.
  • Strong, hands-on experience with LLMs/Transformers, including their theoretical underpinnings and practical application in post-training (e.g., fine-tuning, RLHF, prompting, agent development).
  • Experience deploying backend LLM applications.
  • Proficiency in Python and deep experience with ML/Deep Learning frameworks (e.g., TensorFlow, PyTorch, JAX, HuggingFace, LangChain).

It’d be great if you also had these:

  • Experience with graph-based machine learning techniques.
  • Meaningful experience or a strong understanding of applying AI/ML in another scientific discipline such as biology, psychology, neuroscience, or a related field.
  • Demonstrable research experience, evidenced by contributions to projects, and ideally through publications in relevant ML/NLP venues (e.g., NeurIPS, ICML, ICLR, ACL, EMNLP).
  • Industry experience in a startup or fast-paced R&D environment.

The US base salary range for this full-time position is $307,000 - $422,000 + bonus + equity + benefits. Within the range, individual pay is determined by work location and additional factors, including job-related skills, experience, and relevant education or training. Your recruiter can share more about the specific salary range for your location during the hiring process.

Please note that the compensation details listed in US role postings reflect the base salary only, and do not include bonus, equity, or benefits.

An Equal Opportunity Workplace

At X, we don't just accept difference - we celebrate it, we support it, and we thrive on it for the benefit of our employees, our products and our community. We are 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.

If you have a disability or special need that requires accommodation, please contact us at x-accommodation-request@x.team.

Apply Now

Fields marked with * are required

This position is located in Mountain View, CA. Are you able to commute to the office and/or are you willing to relocate?*
Are you currently or were you previously an Alphabet employee, contractor, or intern?*
Application consent for X

By clicking the “I Accept” button you expressly give your consent for the collection and use of your information as described at https://www.google.com/about/careers/privacy/.  I consent to the processing of my information as described in that policy including that, in limited circumstances, my information may be shared with trusted third parties to assist in certain phases of the hiring process (such as conducting background checks).

If you believe you belong to any of the categories of protected veterans listed below, please indicate by making the appropriate selection. As a government contractor subject to the Vietnam Era Veterans' Readjustment Assistance Act (VEVRAA), we request this information in order to measure the effectiveness of the outreach and positive recruitment efforts we undertake pursuant to VEVRAA. Classification of protected categories is as follows:

A "disabled veteran" is one of the following: a veteran of the U.S. military, ground, naval or air service who is entitled to compensation (or who but for the receipt of military retired pay would be entitled to compensation) under laws administered by the Secretary of Veterans Affairs; or a person who was discharged or released from active duty because of a service-connected disability.

A "recently separated veteran" means any veteran during the three-year period beginning on the date of such veteran's discharge or release from active duty in the U.S. military, ground, naval, or air service.

An "active duty wartime or campaign badge veteran" means a veteran who served on active duty in the U.S. military, ground, naval or air service during a war, or in a campaign or expedition for which a campaign badge has been authorized under the laws administered by the Department of Defense.

An "Armed forces service medal veteran" means a veteran who, while serving on active duty in the U.S. military, ground, naval or air service, participated in a United States military operation for which an Armed Forces service medal was awarded pursuant to Executive Order 12985.

VeteranStatus

Voluntary Self-Identification


For government reporting purposes, we ask candidates to respond to the below self-identification survey. Completion of the form is entirely voluntary. Whatever your decision, it will not be considered in the hiring process or thereafter. Any information that you do provide will be recorded and maintained in a confidential file.

As set forth in X, the moonshot factory’s Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.

Race
Gender