Demo

ML Research Engineer

Exa
San Francisco, CA Full Time
POSTED ON 3/27/2025
AVAILABLE BEFORE 4/26/2025

Transformers feel magical they understand text at near-human level. Yet traditional search engines (cough, Google) don't feel magical they don't understand text at near-human level. It doesn't have to be this way.

Find out more about this role by reading the information below, then apply to be considered.

At Exa, we're training 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. Search is a unique research problem within generative AI instead of generating a single piece of text, search involves asking the same questions about billions of texts.

No other AI lab is exploring this direction. That means if we don't find novel methods for search, they won't be found. Our team has already made breakthroughs not seen in the literature, and we have more coming. Want to explore the search frontier with us?

Desired Experience

  • You have graduate-level ML experience (or are an exceptionally strong undergrad)
  • You can code up a transformer from scratch in pytorch
  • Care about the problem of finding high quality knowledge and recognize how important this is for the world

Example Projects

  • Train a 30B parameter version of our current model
  • 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
  • J-18808-Ljbffr

    If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
    Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

    What is the career path for a ML Research Engineer?

    Sign up to receive alerts about other jobs on the ML Research Engineer career path by checking the boxes next to the positions that interest you.
    Income Estimation: 
    $113,077 - $147,784
    Income Estimation: 
    $135,356 - $164,911
    Income Estimation: 
    $153,902 - $198,246
    Income Estimation: 
    $103,625 - $127,928
    Income Estimation: 
    $88,975 - $120,741
    Income Estimation: 
    $68,121 - $81,836
    Income Estimation: 
    $71,928 - $87,026
    Income Estimation: 
    $125,958 - $157,570
    Income Estimation: 
    $130,030 - $173,363
    Income Estimation: 
    $194,895 - $259,743
    Income Estimation: 
    $192,057 - $260,440
    Income Estimation: 
    $249,515 - $311,938
    Income Estimation: 
    $155,477 - $213,492
    Income Estimation: 
    $125,958 - $157,570
    Income Estimation: 
    $120,989 - $162,093
    Income Estimation: 
    $74,806 - $91,633
    Income Estimation: 
    $71,928 - $87,026
    Income Estimation: 
    $145,337 - $174,569
    Income Estimation: 
    $145,337 - $174,569
    Income Estimation: 
    $153,127 - $203,425
    Income Estimation: 
    $139,626 - $193,276
    Income Estimation: 
    $164,650 - $211,440
    Income Estimation: 
    $130,030 - $173,363
    View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

    Job openings at Exa

    Exa
    Hired Organization Address San Francisco, CA Full Time
    Exa is building an API to the world's knowledge unlike any that have ever existed. Want to design the next-gen of search...
    Exa
    Hired Organization Address San Francisco, CA Full Time
    Exa is building the next generation search engine. That demands completely new UI interfaces. Up for the challenge? All ...
    Exa
    Hired Organization Address California, MO Full Time
    Exa is building an API to the world's knowledge unlike any that have ever existed. Want to design the next-gen of search...
    Exa
    Hired Organization Address San Francisco, CA Full Time
    At Exa, we're on a mission to organize the world’s knowledge by building AI that can answer complex queries with precisi...

    Not the job you're looking for? Here are some other ML Research Engineer jobs in the San Francisco, CA area that may be a better fit.

    Research Engineer, Applied ML

    Recruiting From Scratch, San Francisco, CA

    ML Research Engineer

    Twelve Labs Inc., San Francisco, CA

    AI Assistant is available now!

    Feel free to start your new journey!