Skip to content
ExaExaSan Francisco, CA

Research, ML

ML Research Engineer trains foundational embedding models for web search, designs novel transformer architectures, creates datasets and evals, and beats state-of-the-art performance. Requires graduate-level ML expertise and ability to implement transformers in PyTorch.

150k – 300k/yr
On-siteAI Research

About the role

Responsibilities

  • Dream up novel transformer-based search architectures.
  • Create large-scale datasets and dive deeply into the data.
  • Create evals and beat internal state-of-the-art models.
  • Pre-training: Train hundred billion parameter models.
  • Fine-tuning: Build RLAIF pipelines for search.
  • Implement novel architectures and improve search scores.
  • Build evaluation systems to measure search quality advancements.

Requirements

  • Graduate-level ML experience (or exceptionally strong undergrad).
  • Ability to code a transformer from scratch in PyTorch.
  • Passion for creating high-quality knowledge datasets and solving search problems.

Nice-to-Haves

Compensation/Benefits

  • Premium healthcare (medical, dental, vision).
  • Fertility benefits.
  • Monthly wellness stipend.

Skills

PyTorchTransformersMachine LearningEmbeddingsRlaifPre-TrainingFine-TuningDatasetsEvaluationsRust

Similar roles

AI Research jobs
Distyl AI

Applied AI Researcher, Multi-Agent Systems

Distyl AISan Francisco, CA +1

Develop multi-agent AI architectures for enterprise coordination and collaborative reasoning. Requires research experience in MARL/GNNs, strong prototyping skills, and daily AI tool usage.

150k – 250k/yr
HybridAI Research
Basis

Intelligence Architect

BasisNew York, NY

Designs and authors context, procedures, skills, and system prompts for AI agents performing autonomous accounting tasks. Ensures consistency, monitors performance, and shapes future capabilities through precise language and systems thinking.

150k – 225k/yr
On-siteAI Research
AfterQuery

Research Scientist - Frontier Data

AfterQuerySan Francisco, CA

Designs datasets and evaluation frameworks for frontier AI models, collaborating with top labs to expose failure modes, refine RLHF/RLVR pipelines, and measure data impact on capabilities. Requires strong quantitative skills, familiarity with LLM training, and research experience up to master's level.

150k – 250k/yr
On-siteAI Research
AfterQuery

Research Scientist - Post Training

AfterQuerySan Francisco, CA

Design and run SFT/RL experiments to measure dataset impact on LLM performance, capabilities, and alignment. Collaborate with labs to provide evidence of improvements; requires strong LLM training knowledge and fast experimentation, ideally undergrad/master's research background.

150k – 250k/yr
On-siteAI Research
Astera

Machine Learning Researcher - Springtail

AsteraEmeryville, CA

Conducts fundamental research on data-efficient ML architectures, including bootstrapped program synthesis and self-synthesizing learning systems. Requires Master's in ML/math, PyTorch fluency, and research experience.

150k – 300k/yr
HybridAI Research