Head of ML/AI Exploration, Architecture, & Engineering

Head of ML/AI Exploration, Architecture, & Engineering

Join to apply for the Head of ML/AI Exploration, Architecture, & Engineering role at X, The Moonshot Factory

Head of ML/AI Exploration, Architecture, & Engineering

Join to apply for the Head of ML/AI Exploration, Architecture, & Engineering role at X, The Moonshot Factory

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 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 engineering 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 engineering 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 $248,000 - $345,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.

Seniority level

• Seniority level

  Mid-Senior level

Employment type

• Employment type

  Full-time

Job function

• Job function

  Engineering and Information Technology

• Industries

  Research

Referrals increase your chances of interviewing at X, The Moonshot Factory by 2x

Get notified about new Head of Machine Learning jobs in Mountain View, CA .

San Jose, CA $165,300.00-$327,600.00 1 week ago

San Francisco Bay Area $222,000.00-$299,000.00 2 weeks ago

Director of Engineering, ML Infrastructure

Mountain View, CA $222,775.00-$333,925.00 2 weeks ago

Head of Computer Vision and Machine Learning

Redwood City, CA $160,000.00-$220,000.00 1 year ago

Sr Director or Director, Machine Learning Engineering, Agent Factory

Pleasanton, CA $259,200.00-$460,800.00 1 day ago

San Jose, CA $232,000.00-$398,750.00 1 week ago

Sunnyvale, CA $235,000.00-$400,000.00 1 week ago

Director, ADAS Perception Software Engineering and Machine Learning

Palo Alto, CA $200,000.00-$250,000.00 1 month ago

Head of Product, Computer Vision and Artificial Intelligence

Menlo Park, CA $253,000.00-$314,000.00 2 weeks ago

Director, Machine Learning, Discovery Chemistry

Palo Alto, CA $220,000.00-$270,000.00 2 hours ago

Director of Data Science - Recommendations and E-Commerce

Pleasanton, CA $201,700.00-$267,300.00 2 weeks ago

Mountain View, CA $272,900.00-$400,000.00 1 week ago

Director of Finance and Administration, Biomedical Data Science

Stanford, CA $205,050.00-$283,688.00 2 weeks ago

Director of Finance and Administration, Biomedical Data Science

Stanford, CA $205,050.00-$283,688.00 7 months ago

Mountain View, CA $192,000.00-$270,000.00 2 weeks ago

Head of Autonomous Driving Technology & AI Development

Sunnyvale, CA $290,000.00-$450,000.00 1 week ago

Director, Scientific Content Engineering/Data Science, QDI

Associate Director, People Analytics Insight Partner

Foster City, CA $210,375.00-$272,250.00 1 day ago

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.

#J-18808-Ljbffr