About the job
At Exa, we are revolutionizing the landscape of search engines by constructing a robust platform from the ground up to cater to every AI application. Our team is dedicated to building a colossal infrastructure that efficiently crawls the web, trains cutting-edge embedding models for indexing, and develops high-performance vector databases using Rust for swift searching capabilities. We are proud to operate a $5M H200 GPU cluster that powers tens of thousands of machines with unparalleled efficiency.
As a member of our Machine Learning team, you will play a pivotal role in training foundational models for advanced search functionalities. Our mission is to create systems capable of instantaneously filtering the vast knowledge available online to deliver exactly what you need, no matter the complexity of your query. Essentially, we aim to transform the web into an extraordinarily powerful database.
We are in search of a passionate Machine Learning Research Engineer who will focus on training embedding models to enhance web search accuracy. This role will involve conceptualizing innovative transformer-based architectures for search, generating extensive datasets, conducting evaluations, and consistently striving to surpass our internal state-of-the-art benchmarks.

