Apply on Employer Site

X, The Moonshot Factory · 2 weeks ago

Applied ML Research Lead, X, Early Stage Supply Chain Project

Mountain View, CA

Full-time

Onsite

Lead/Staff

$248K/yr - $330K/yr

10+ years exp

X, The Moonshot Factory is focused on revolutionizing the global supply chain through AI innovation. The Applied ML Research Lead will play a pivotal role in designing, architecting, and deploying advanced AI systems while leading an engineering team and bridging the gap between AI research and enterprise applications.

AerospaceAugmented RealityAutomotiveRobotics

Responsibilities

Develop and deliver high-quality, production-grade AI enterprise software, from ideation and architectural design to development, testing, deployment, and ongoing maintenance in complex enterprise environments

Help drive and execute key decisions on software architecture and features

Develop experiences that inspire confidence, delight and resonate with end users, and encourage adoption and trust

Enhance AI reasoning by innovating on how Foundation Models are utilized, improving internal workflows, and developing/integrating specialized tools

Explore, apply, and innovate state-of-the-art LLM and machine learning techniques

Lead and mentor technical initiatives: Guide project execution, provide technical leadership, and cultivate a strong, high-performing team

Ensure responsible AI development: Implement robust safety principles, including sandboxing and traceability of self-modifications, to mitigate risks and ensure alignment with business objectives and ethical guidelines

Qualification

Applied ML researchProduction-grade AI softwareDeep learning frameworksMLOpsPython developmentEnterprise software experienceCloud platformsTechnical leadershipCommunication skillsOrganization skills

Required

MS/PhD degree or equivalent practical experience in computer science, engineering or a related scientific or quantitative field

10+ years experience conducting applied ML research and high-performance implementations of machine learning algorithms and translating research concepts or novel techniques into robust, production-ready systems

7+ years designing, developing, maintaining, and releasing production-grade enterprise/business software

4+ years experience as a technical lead for small team

Extensive applied research experience with: Applying LLMs and agentic AI, particularly with techniques like RAG or knowledge graphs

Achieving strong results with sophisticated model prompting

MLOps, including model deployment, versioning and performance monitoring in production environments

Innovating ways to improve model accuracy and quantify drift, overfitting and regression

Deep learning frameworks like PyTorch, Tensorflow, JAX

Optimizing system/model performance (e.g. speed, cost, throughput)

Python development, with experience in the latest Python toolchains and frameworks, including developing for various platforms (e.g., full-stack web applications, backend services, data pipelines)

Extensive enterprise software experience with: Using LLMs to optimize software development processes and workflows

Collaborative software development workflows and platforms including GitHub, agile development/scrum, unit testing, CI/CD, and production operations

Integrating AI into enterprise software environments, including Large Language Models (LLMs), Foundation Models, multi-agent system architectures and their orchestration agentic behaviors, tool use, and reasoning capabilities

Integrating software with complex enterprise legacy systems, including ERP integration (e.g., SAP, Oracle)

Deploying and managing applications on major cloud platforms (e.g., Google Cloud Platform, AWS, Azure) and an understanding of underlying infrastructure primitives (e.g., Kubernetes, Terraform)

Experience working in a startup or equivalent fast-paced environment, able to thrive and execute quickly in an environment of change and uncertainty

Excellent written and verbal communication skills, adept at explaining complex technical concepts to both technical and non-technical audiences in a clear and concise manner

Excellent organization and planning skills

Preferred

Experience with supply chain-related enterprise software applications for procurement, sourcing, product lifecycle management, or supply chain management

Experience with Google's AI tools and platforms

Experience with Google Cloud and Workspace

Research papers or other public-facing documents or presentations that demonstrate technical thought leadership