Staff Machine Learning Engineer - Applied AI

United States Digital Space LLC; San Francisco (CA); On-site; USD 232,000 - 258,000; Full time

Job summary

United States Digital Space LLC in San Francisco is seeking a Staff ML Engineer (IC6) to define and lead the foundation model strategy powering our discovery experiences. You will shape technical direction across teams and influence product strategy while delivering measurable impact at global scale.

You will own end-to-end ML strategy for foundation models across search, recommendations, and conversational AI, mentor engineers, and drive cross-team initiatives spanning retrieval, ranking,

Qualifications

• Masters degree or Ph.D in Computer Science, Engineering, Mathematics.

Responsibilities

• Own the end-to-end technical strategy for foundation models across Search, Recommendations, and Conversational AI.

Skills

ML expertise

Transformers

PyTorch

Distributed training

Product intuition

Leadership across teams

Education

Masters/PhD in Computer Science, Engineering, Mathematics

Job description

About the Team

The Applied AI team collaborates with product teams across the company to deliver innovative AI solutions for core business problems. We work closely with engineering, product and data science teams to understand core business problems and the potential for AI solutions, then deliver those AI solutions end-to-end. Key areas of expertise include Personalization, Generative AI, Computer Vision, ML Optimization and Geospatial AI.

About the Role

We are building AI-native discovery experiences across Mobility and Delivery. Search, recommendations, and conversational AI are central to how millions of users discover rides, restaurants, grocery items, and retail products every day. We are hiring a Staff ML Engineer (IC6) to define and lead the foundation model strategy powering these experiences. At this level, you will not just build models - you will shape technical direction across teams, influence product strategy, and deliver measurable impact at global scale.

What the Candidate Will Do

• Own the end-to-end technical strategy for foundation models across Search, Recommendations, and Conversational AI.
• Drive architecture decisions that influence multiple product surfaces (Eats, Grocery, Retail, Mobility).
• Lead cross-team initiatives spanning Retrieval, Ranking, Personalization, and LLM-powered assistants.
• Define long-term investment areas (build vs fine-tune vs partner models).
• Mentor senior engineers and act as a technical multiplier across the org.

Basic Qualifications

• Masters degree or Ph.D in Computer Science, Engineering, Mathematics
• 8+ years of ML experience, including significant work on large-scale deep learning systems.
• Demonstrated ownership of high-impact ML systems in search, recommendations, or conversational AI.
• Deep expertise in transformers, retrieval systems, ranking, and embedding architectures.
• Strong experience with PyTorch and distributed training .
• Track record of influencing technical direction across teams.
• Strong product intuition and ability to connect model improvements to business outcomes.

Preferred Qualifications

• Experience leading multi-team ML initiatives.
• Defined long-term technical roadmaps adopted across orgs.
• Elevated engineering standards through mentorship and technical leadership.

For San Francisco, CA-based roles: The base salary range for this role is USD $232,000 per year - USD $258,000 per year.For Seattle, WA-based roles: The base salary range for this role is USD $232,000 per year - USD $258,000 per year.For Sunnyvale, CA-based roles: The base salary range for this role is USD $232,000 per year - USD $258,000 per year.For all US locations, you will be eligible to participate in the company's bonus program, and may be offered an equity award & other types of comp. All full-time employees are eligible to participate in a 401(k) plan. You will also be eligible for various benefits.Ready to Ride?This isn't the kind of place where you follow a playbook - it's where you help write one. If you're driven by impact, energized by challenge, and ready to shape how the world moves - we'd love to hear from you.You may be eligible for bonuses, equity, and other compensation, as well as a range of benefits. Explore our benefits.Offices remain key to collaboration and the company's culture. Unless approved for full remote work, employees must spend at least 50% of their time in-office. Some roles, like those at greenlight hubs, require full-time in-office presence. Ask your Recruiter for details about this role's requirements.the company is proud to be an Equal Opportunity employer. All qualified applicants will receive consideration for employment without regard to sex, gender identity, sexual orientation, race, color, religion, national origin, disability, protected Veteran status, age, or any other characteristic protected by law. 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 let us know by completing this form.