ML Research Engineer: Large-Scale Embeddings for Search

Exa; San Francisco (CA); On-site; USD 180,000 - 240,000; Full time

Benefits offered by this job

Premium healthcare benefits

Fertility benefits

16 weeks parental leave

Wellness stipend

Job summary

Exa in San Francisco seeks an ML Research Engineer to train embedding models for perfect web search. You will design transformer-based architectures, build large-scale datasets, and create robust evals to push the state-of-the-art. This role requires a strong ML background and the drive to advance search quality.

Location is in-person in San Francisco, with visa sponsorship for international candidates. Exa offers premium healthcare, fertility benefits, 16 weeks of parental leave, and a monthly

Qualifications

• Graduate-level ML experience required (MS/PhD or exceptional undergrad).
• Ability to implement a transformer from scratch in PyTorch.
• Experience creating large-scale datasets and analyzing them.
• Interest in high quality knowledge for search and information retrieval.

Responsibilities

• Train embedding models for web-scale search.
• Dream up transformer-based architectures for search tasks.
• Create datasets and evaluation protocols to measure progress.
• Beat internal state-of-the-art models and iterate toward better accuracy.

Skills

Graduate-level ML experience

Transformer in PyTorch

Large-scale datasets

Education

Master's or PhD in ML

Tools

PyTorch

Job description

Exa in San Francisco seeks an ML Research Engineer to train embedding models for perfect web search. You will design transformer-based architectures, build large-scale datasets, and create robust evals to push the state-of-the-art. This role requires a strong ML background and the drive to advance search quality.

Location is in-person in San Francisco, with visa sponsorship for international candidates. Exa offers premium healthcare, fertility benefits, 16 weeks of parental leave, and a monthly