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