Latest AI JobsBrowse Jobs
E

Research, ML

ExaSan Francisco, CaliforniaResearch
Apply on company site
Salary
$180K - $350K USD
Job type
Full-time
Posted
26 Jun 2025
Status
Open · checked 7d ago

About this role

Role Overview

The Research ML Engineer position at Exa is a full-time, in-person role located in San Francisco, California. This opportunity suits professionals with graduate-level machine learning experience who are eager to build massive-scale ML systems and train foundational models for search.

Key Responsibilities

Design and experiment with novel transformer-based search architectures.
Create large-scale datasets and conduct deep data analysis.
Build robust evaluation systems to measure search quality advancements.
Continuously iterate to outperform internal state-of-the-art models.
Undertake projects such as pre-training large models, building RLAIF pipelines for search, and developing innovative search architectures.

Requirements

Possess graduate-level machine learning experience, defined as a Master's degree or PhD with at least two years of relevant experience, or status as an exceptionally strong undergraduate.
Demonstrate the ability to code a transformer entirely from scratch using PyTorch.
Show a strong interest in creating large-scale datasets and exploring data deeply.
Value the importance of discovering high-quality knowledge to impact the world positively.
Be available to work in person at the San Francisco office.

Skills and Tools

PyTorch
Transformer-based architectures
Large-scale dataset creation and curation
Machine learning research and evaluation design

Salary and Benefits

Annual salary ranging from $180,000 to $350,000 USD.
Premium healthcare coverage including medical, dental, and vision benefits.
Fertility benefits and a monthly wellness stipend for all employees.
Sixteen weeks of fully paid parental leave for all new parents.
Visa sponsorship support for international candidates, including STEM OPT, OPT, H1B, O1, and E3 options.

About the Company

Exa operates as an applied AI lab focused on creating a revolutionary search engine designed to surpass traditional platforms. The company crawls the web at massive scale, trains state-of-the-art embedding models, and designs high-performance vector databases. Backed by funding from Lightspeed, Benchmark, and a16z, Exa currently powers search capabilities for hundreds of thousands of developers and major tech platforms alike.

How to Apply

Interested candidates should visit the official company careers page to submit their application.

Skills and tech stack

Questions candidates ask

Where is this job located?

This is an in-person role located in San Francisco, California.

What is the salary range for this position?

The salary ranges from $180,000 to $350,000 USD per year.

What experience is required to apply?

Candidates need graduate-level machine learning experience, which includes a Master's degree or PhD with at least two years of relevant experience, or being an exceptionally strong undergraduate.

Does Exa provide visa sponsorship?

Yes, Exa sponsors international candidates for visas such as STEM OPT, OPT, H1B, O1, and E3.

What benefits are included with employment?

Benefits include premium healthcare, fertility benefits, a monthly wellness stipend, and 16 weeks of fully paid parental leave.

Ready to apply?

The application is on the official Exa careers page.

Apply on company site