Research, ML

Exa is an applied AI lab building a search engine unlike the world has ever seen. 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

Requirements

Graduate-level ML experience (Master's degree or PhD with at least 2 years of relevant experience, or are an exceptionally strong undergrad); You can code up a transformer from scratch in PyTorch; You like creating large-scale datasets and diving deeply into the data; You 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 (e.g., STEM OPT, OPT, H1B, O1, E3)