# Research, ML

**Company:** [Exa](https://hotfix.jobs/companies/exa)
**Location:** San Francisco, CA
**Role:** AI Research
**Salary:** $150k – $300k/yr
**Skills:** PyTorch, Transformers, Machine Learning, Embeddings, Rlaif, Pre-Training, Fine-Tuning, Datasets, Evaluations, Rust
**Posted:** 2025-06-26

> ML Research Engineer trains foundational embedding models for web search, designs novel transformer architectures, creates datasets and evals, and beats state-of-the-art performance. Requires graduate-level ML expertise and ability to implement transformers in PyTorch.

## Job Description

## Responsibilities
- Dream up novel transformer-based search architectures.
- Create large-scale datasets and dive deeply into the data.
- Create evals and beat internal state-of-the-art models.
- Pre-training: Train hundred billion parameter models.
- Fine-tuning: Build RLAIF pipelines for search.
- Implement novel architectures and improve search scores.
- Build evaluation systems to measure search quality advancements.

## Requirements
- Graduate-level ML experience (or exceptionally strong undergrad).
- Ability to code a transformer from scratch in PyTorch.
- Passion for creating high-quality knowledge datasets and solving search problems.

## Nice-to-Haves

## Compensation/Benefits
- Premium healthcare (medical, dental, vision).
- Fertility benefits.
- Monthly wellness stipend.

## Similar roles

- [Applied AI Researcher, Multi-Agent Systems](https://hotfix.jobs/jobs/6aa59b8a-03f0-4837-bb6e-dd10d59ce3c7) - Distyl AI - San Francisco, CA - $150k – $250k/yr
- [Intelligence Architect](https://hotfix.jobs/jobs/86d5c45a-0c28-428a-9710-53f4bbe379d4) - Basis - New York, NY - $150k – $225k/yr
- [Research Scientist - Frontier Data](https://hotfix.jobs/jobs/838be600-594a-404f-aa4b-5bdb573721c4) - AfterQuery - San Francisco, CA - $150k – $250k/yr
- [Research Scientist - Post Training](https://hotfix.jobs/jobs/69d58c6c-4776-40c5-8cf8-6da6cd0650fe) - AfterQuery - San Francisco, CA - $150k – $250k/yr
- [Machine Learning Researcher - Springtail](https://hotfix.jobs/jobs/7beb0953-43ae-4c56-8de8-5a505d2538c8) - Astera - Emeryville, CA - $150k – $300k/yr

**Apply:** https://hotfix.jobs/jobs/aa12eba3-3ec5-487e-b31a-3b98e24b5dfb
**Canonical:** https://hotfix.jobs/jobs/aa12eba3-3ec5-487e-b31a-3b98e24b5dfb