Research, ML
ExaSwitzerland7 hours ago
Full-timeResearch, Analyst +1
Track This Job
Add this job to your tracking list to:
- Monitor application status and updates
- Change status (Applied, Interview, Offer, etc.)
- Add personal notes and comments
- Set reminders for follow-ups
- Track your entire application journey
Save This Job
Add this job to your saved collection to:
- Access easily from your saved jobs dashboard
- Review job details later without searching again
- Compare with other saved opportunities
- Keep a collection of interesting positions
- Receive notifications about saved jobs before they expire
AI-Powered Job Summary
Get a concise overview of key job requirements, responsibilities, and qualifications in seconds.
Pro Tip: Use this feature to quickly decide if a job matches your skills before reading the full description.
Exa is building a search engine from scratch to serve every AI application. We build massive-scale infrastructure to crawl the web, train state-of-the-art embedding models to index it, and develop super high performant vector databases in Rust to search over it. We also own a $5M H200 GPU cluster that regularly lights up tens of thousands of machines.
On the ML team, 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. Basically, put the web into an extremely powerful database.
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.
Desired Experience
Compensation Range: CHF 120K - CHF 200K
On the ML team, 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. Basically, put the web into an extremely powerful database.
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.
Desired Experience
- You have graduate-level ML 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
- Pre-training: Train a hundred billion parameter model
- Fine-tuning: Build an RLAIF pipeline for search
- Dream up a novel architecture for search in the shower, then code it up and beat our best model's top score
- Build an eval system that answers how do we know we're advancing our search quality? (this is an incredibly difficult question to answer)
Compensation Range: CHF 120K - CHF 200K
Key Skills
Ranked by relevanceReady to apply?
Join Exa and take your career to the next level!
Application takes less than 5 minutes