Skip to content
ExaExaSan Francisco, CA

Research Engineer, Generalist

Generalist Research Engineer working across Exa's search and retrieval stack including crawling, parsing, ML performance, and retrieval algorithms to improve search quality and performance for customers.

180k – 350k
On-siteML Engineering

About the role

Responsibilities

  • Design and ship a new parser that dramatically improves retrieval performance
  • Improve throughput and latency of ML models
  • Make our web crawler 3x faster
  • Work across the search and retrieval stack: crawling, parsing, ML performance, retrieval algorithms
  • Improve end quality of search endpoints for customers

Requirements

  • High attention to detail; ability to simultaneously understand customer needs and implementation details of the system
  • Great intellectual horsepower - generalists who are eager to go deep
  • Excited to work across search and retrieval stack: crawling, parsing, ML performance, retrieval algorithms and more
  • Comfortable working across different systems, and shipping fast
  • Care about the problem of finding high quality information and recognize how important this is for the world

Nice-to-Haves

  • Experience building massive-scale ML systems

Compensation

  • Salary range: $180,000 - $350,000

Skills

Machine LearningCrawlingParsingRetrieval AlgorithmsVector DatabasesEmbedding Models

Similar roles

ML Engineering jobs
Notion

Software Engineer, AI Platform

NotionSan Francisco, CA +1

Build and scale the shared AI platform foundations at Notion, enabling fast and safe shipping of AI products. Requires experience with LLM/ML platforms, strong ownership, and comfort across backend, infrastructure, and product code.

180k – 201k
Hybrid5+ YOEML Engineering
Baseten

Software Engineer - BIS

BasetenSan Francisco, CA

As a Software Engineer on the Inference Stack team, you will build the distributed runtime that powers large-scale LLM inference. This role involves working across the stack, from developer experience to low-level infrastructure, and owning systems in production.

180k – 360k
HybridML Engineering
Build

AI Engineer (Core)

BuildSan Francisco, CA

Builds core infrastructure for production AI agents including runtime, evaluation systems, retrieval, tool orchestration, observability, and reliability features for high-stakes real estate workflows. Requires strong systems engineering with Python, backend, and LLM experience.

180k – 250k
On-siteML Engineering
Lightning AI

Research Engineer

Lightning AINew York, NY +1

Develops performance optimizations for ML models across graph, kernel, and system levels using PyTorch and Thunder compiler. Builds tools, collaborates with partners, and contributes to open-source while requiring strong PyTorch expertise and optimization experience.

180k – 250k
RemoteML Engineering
Rillet

Applied AI Engineer

RilletSan Francisco, CA +1

Designs and ships production AI systems including agentic workflows, RAG pipelines, and LLM integrations for an AI-native ERP platform serving finance teams. Requires 3+ years backend experience and 2+ years production AI with Python proficiency.

180k – 240k
Hybrid3+ YOEML Engineering