Skip to content

Research, ML

150k – 300kSan Francisco, CAAI ResearchOnsite
Summary

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.

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 at this salary range
All AI Research jobs →
Snowflake

AI Research Scientist, New Grad – Agents & Reinforcement Learning

Conduct research on autonomous AI agents and reinforcement learning to build self-improving systems that reason, code, and learn at scale within the Snowflake Data Cloud. Requires a PhD (or equivalent) and strong expertise in RL and agentic AI.

176k – 230kBellevue, WAAI ResearchOn-siteEntry levelJAXDPO
Together AI

Frontier Agents Intern

Research intern on the Agents team building and aligning frontier AI systems for complex agentic and scientific tasks. Focus on post-training methods, evaluation frameworks, self-learning, and scalable agent infrastructure.

121k – 131kSan Francisco, CAAI ResearchOn-siteEntry levelJAXNLP
Snowflake

Post-Doctoral Researcher

Post-doctoral researcher conducting independent and collaborative AI/ML research focused on high-impact domains like medicine, finance, and law. Requires a recent or imminent PhD and publications in top venues.

160k – 220kBellevue, WAAI ResearchHybridEntry levelJAXRAG
SpotOn

Senior Software Engineer - Python/Typescript

Senior engineer building AI-driven automation systems to replace manual business workflows across operations, sales, and support. Requires 7+ years experience, production Python/TypeScript skills, and 1-2 years building agentic AI systems.

160k – 190kChicago, IL +3AI ResearchHybrid7+ YOEAWSLLMs
Datology AI

Research Engineer

As a Research Engineer, you will conduct and enable cutting-edge research, translating it into the core product pipeline. You will develop and improve state-of-the-art data curation strategies, accelerating research and ensuring product innovation.

180k – 300kRedwood City, CAAI ResearchOn-site4+ YOEML ModelsAI Models