Research, ML

Exa is an applied AI lab building a search engine unlike the world has ever seen. We build massive-scale infra to crawl the entire web, train state-of-the-art embedding models to process it, and design super high performant vector databases to retrieve over it. We now power search for Cursor, Cognition, HubSpot, and over 400,000 developers and have raised $350m from Lightspeed, Benchmark, and a16z. Our ultimate goal is to build perfect search over all the world's information, far beyond Google. The ML organization sits at the heart of our mission. We train foundational models for search. Our goal is to build systems that can instantly filter the world's knowledge to exactly what you want, no matter how complex your query. We're looking for an ML Research Engineer to train embedding models for perfect search over the web. The role involves dreaming up novel transformer-based search architectures, creating datasets, creating evals, beating our internal SoTA, and repeat.

Responsibilities

Train embedding models for perfect search over the web; dream up novel transformer-based search architectures; create datasets; create evals; beat internal SoTA; repeat

Requirements

graduate-level ML experience (or exceptionally strong undergrad); code up a transformer from scratch in PyTorch; like creating large-scale datasets and diving deeply into the data; care about the problem of finding high quality knowledge and recognize how important this is for the world

Benefits

premium healthcare benefits (medical, dental, vision); fertility benefits; 16 weeks of fully paid parental leave for all new parents; monthly wellness stipend; visa sponsorship