Join Exa as an ML Research Engineer to develop cutting-edge embedding models for a revolutionary search engine.
Posted by employer 1 month ago
First seen on Joblaze 1 week ago
Last verified on the company career page 1 day ago
Skills & Technologies
Role intensity
70% hands-on coding
AI in the day-to-day
We train foundational models for search and build systems that filter the world's knowledge.
Requirements
- Experience
- 1+ years
- Education
- Master's degree
- Visa
- Sponsorship available
Not disclosed in this posting: compensation.
Benefits
Joblaze summary
In this role, the ML Research Engineer at Exa focuses on developing and training advanced embedding models to enhance search capabilities across the web. Proficiency in PyTorch and experience with large-scale dataset creation are essential, as the position involves innovating transformer-based architectures and evaluating their performance. This opportunity is ideal for individuals with a strong academic background in machine learning, particularly those eager to tackle complex challenges in information retrieval. Exa's mission to redefine search technology places the ML team at the forefront of applied AI innovation.
Joblaze insights
- Listed 9 days ago on Joblaze — tracked directly from Exa's career page.
- Exa has 54 other open roles including Design Engineer, Lifecycle Marketing Manger, Sales Development Representative, SF.
- 799 active Machine Learning roles on Joblaze right now.
- 481 active PyTorch roles on Joblaze right now.
- 6 active Transformer roles on Joblaze right now.
- Typical Machine Learning salary we track runs $145,000–$230,000 (median $185,000, across 356 roles). See Machine Learning salary trends
Quick facts
- Is the Research, ML role remote?
- No — this is an on-site role in Singapore.
- How much experience is required?
- At least 1 year of relevant experience for this Research, ML role.
- Where is the role based?
- Exa is hiring for this position in Singapore.
- What's the tech stack?
- Joblaze extracted these technologies from the posting: Machine Learning, PyTorch, Transformer, embedding models.
- Does Exa sponsor work visas for this role?
- Yes — the posting indicates visa sponsorship is available for the right candidate.
- What seniority level is this role?
- Exa targets mid-level candidates for this position.
- Is this full-time or contract?
- Full-time for this Research, ML role at Exa.
From the original posting
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. If you want to build massive-scale ML systems that will define the way the new AI world consumes information, this is the place for you.
Research at Exa
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. 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
Example Projects
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)
Logistics
Location: This is an in-person opportunity in Singapore.
Visas: We’re happy to sponsor international candidates! While we cannot guarantee your visa, we have historically been successful in sponsoring candidates from all over the world. If you receive an offer, our team will work hard to get you a visa.
Benefits: We offer premium healthcare benefits (medical, dental, vision), fertility benefits, 16 weeks of fully paid parental leave for all new parents, and a monthly wellness stipend to all of our employees.
Exa is an equal opportunity employer. All qualified applicants will receive consideration for employment without regard to race, creed, color, religion, sex, sexual orientation, gender identity or expression, national origin, disability, age, veteran status, marital status, pregnancy or related conditions, criminal histories consistent with applicable law, or any other basis protected by applicable law.