Research Scientist / Research Engineer - Foundations, Search

Skills

About the Role

You will work on search efforts, designing new embedding training objectives, scalable vector store architectures, and dynamic indexing methods. You will build embedding models and retrieval systems, drive innovation in dense, sparse, and hybrid representations, and work with Pretraining, Inference and other Research teams to integrate retrieval across the model lifecycle. The role is based in San Francisco with a hybrid model of 3 days in the office per week.

Requirements

• Proven experience leading high-performance teams of researchers or engineers in ML infrastructure or foundational research
• Deep expertise in representation learning, embedding models, or vector retrieval systems
• Familiarity with transformer-based LLMs and embedding spaces
• Research experience in contrastive learning, embedding learning, or metric learning
• Track record of building or scaling large ML systems, particularly embedding pipelines

Responsibilities

• Tackle embedding models and retrieval systems optimized for grounding, relevance, and adaptive reasoning
• Build end-to-end infrastructure for training, evaluating, and integrating embeddings into frontier models
• Drive innovation in dense, sparse, and hybrid representation techniques, metric learning, and learning-to-retrieve systems
• Collaborate with Pretraining, Inference, and other Research teams to integrate retrieval throughout the model lifecycle
• Contribute to long-term vision of AI systems with memory and knowledge access

Benefits

• Equity
• Medical, dental, and vision insurance with HSA employer contributions
• Health FSA, Dependent Care FSA, and commuter pre-tax accounts
• 401(k) with employer match
• Paid parental leave (up to 24 weeks birth parents, 20 weeks non-birthing parents)
• Paid medical and caregiver leave (up to 8 weeks)
• Flexible PTO for exempt employees; up to 15 days for non-exempt
• 13+ paid company holidays and company office closures
• Paid sick or safe time
• Mental health and wellness support
• Employer-paid basic life and disability coverage
• Annual learning and development stipend
• Daily meals in offices and meal delivery credits
• Relocation support
• Charitable donation matching and wellness stipends
• Performance-related bonus