Research, ML
San Francisco, California$180k - $350k + equityFull-timePosted 11mo ago
About the role
ML Research Engineer to train embedding models for search over the web, develop novel transformer-based architectures, create datasets and evals, and push state-of-the-art performance for Exa’s large-scale search system.
Responsibilities
- Train embedding models for search over the web.
- Design novel transformer-based search architectures.
- Create large-scale datasets and evaluation protocols.
- Beat internal state-of-the-art models and iterate on architectures and datasets.
- Build an evaluation system to measure improvements in search quality.
Qualifications
- Graduate-level ML experience (Master's degree or PhD with at least 2 years of relevant experience, or exceptionally strong undergrad).
- Ability to code up a transformer from scratch in PyTorch.
- Experience creating large-scale datasets and deeply analyzing data.
- Interest in solving the problem of finding high-quality knowledge for the world.
Skills
Required:Transformer modelsPyTorch
Benefits
- premium healthcare benefits (medical, dental, vision)
- fertility benefits
- 16 weeks of fully paid parental leave for all new parents
- monthly wellness stipend
Last updated Aug 26, 2026. Interested? Apply on the company site.
About Exa
Web search rebuilt for LLMs
Exa is a search engine built for AIs. We enable developers to filter the internet's content with complex queries in order to bring the world's knowledge into their AI apps.
ai-infrastructure-and-developmentapisartificial-intelligencebrex-fastest-growing-2025searchwing-list-2026wing-list-2026-mid-stage
More open roles at Exa
- Head of Marketing, EMEA1w agoLondon · Marketing
- Head of Marketing, APJ2w agoSydney · Marketing
- Research Engineer, Index Intelligence2w agoSan Francisco, California · Research
- Research Engineer, Content Understanding2w agoSan Francisco, California · Research
- Commercial Counsel3w agoNew York / San Francisco · Legal