Apply on Employer Site

X, The Moonshot Factory · 9 hours ago

Head of ML/AI Exploration, Architecture, & Engineering

Mountain View, CA

Full-time

Onsite

Director/Executive

$248K/yr - $345K/yr

X, The Moonshot Factory is Alphabet's innovative hub for developing world-changing ideas. They are seeking a Head of ML/AI Exploration, Architecture, & Engineering to lead a team in applying cutting-edge AI to solve complex problems across various projects.

AerospaceAugmented RealityAutomotiveRobotics

Responsibilities

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

Qualification

PhD in Computer ScienceML/AI techniquesLLMs/TransformersPythonML/Deep Learning frameworksGraph-based ML techniquesResearch experienceIndustry experienceMentoring

Required

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)

Preferred

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