Skip to content
SnowflakeSnowflakeBellevue, WA

Staff Research Scientist, Exotic AI

Build next-generation training infrastructure for physical AI models that perceive, reason, and act in structured environments. Lead development of representation models, latent world models, and policy optimization systems.

236k – 339k
On-site8+ YOEAI Research

About the role

Responsibilities

  • Design and build scalable training infrastructure for representation models (e.g., contrastive and self-supervised approaches like CLIP/SigLIP, DINO/MAE, and joint-embedding predictive architectures)
  • Develop latent world models that learn environment dynamics through imagined rollouts, enabling model-based reasoning and planning (Dreamer-style, I-JEPA/V-JEPA families)
  • Architect and implement action/policy model pipelines, including vision-language-action models and diffusion-based policy learning
  • Build generative simulator frameworks that produce controllable, physically plausible future states (video world models in the spirit of Cosmos/Genie/Sora)
  • Develop multimodal generative model capabilities that fuse visual, language, and structured inputs for downstream reasoning and decision-making
  • Lead cross-team technical decisions on training frameworks, data pipelines, and model evaluation infrastructure
  • Drive research-to-production pathways, translating prototype systems into reliable, performant platform capabilities
  • Contribute to the broader research community through publications, open-source releases, and collaboration with academic partners

Requirements

  • 8+ years of relevant experience in machine learning engineering, AI research, or a closely related field (or equivalent experience)
  • Deep expertise in at least two of the following: representation learning, world models, reinforcement learning, generative modeling, robotics/embodied AI, or scientific ML
  • Hands-on experience training large-scale models (vision, language, or multimodal) with distributed compute
  • Strong software engineering fundamentals: system design, performance optimization, and production-quality code
  • Demonstrated ability to drive cross-team technical initiatives with ambiguity and limited direction
  • Track record of translating research ideas into working systems at scale
  • MS or Ph.D. in Computer Science, Machine Learning, Robotics, Physics, or a related field, or equivalent experience

Nice-to-Haves

  • Experience with latent dynamics modeling, model-based RL, or physics-informed neural networks (GraphCast, FourCastNet, AlphaFold-style architectures)
  • Contributions to open-source ML frameworks or foundation model training codebases
  • Background in scientific/structured models (molecular modeling, materials science, weather/climate)
  • Experience building controllable video generation or neural simulation environments
  • Publications at top venues (NeurIPS, ICML, ICLR, CVPR, CoRL, RSS)

Skills

PyTorchTensorFlowJAXRepresentation LearningWorld ModelsReinforcement LearningGenerative ModelingRoboticsVision-Language ModelsDiffusion ModelsDistributed TrainingModel Evaluation

Similar roles

AI Research jobs
Snowflake

Staff Applied Research Scientist

SnowflakeMenlo Park, CA

Lead applied research applying formal methods, automated reasoning, and AI (incl. LLMs) to improve correctness, reliability and velocity of Snowflake's cloud data platform and distributed systems. Translate ideas into production capabilities; requires PhD + 8+ years experience and track record of impact.

236k – 339k
On-site8+ YOEAI Research
Snowflake

Staff Research Scientist, AI Agents & LLMs

SnowflakeBellevue, WA +1

Leads research in agentic AI and LLMs, developing models for enterprise reasoning, autonomous agents with tool use, and production systems. Requires PhD, expertise in LLM training/fine-tuning, agent systems, and technical leadership.

236k – 339k
HybridAI Research
Writer

Staff AI Research Scientist

WriterSan Francisco, CA +2

Lead high-impact research on LLMs and agentic systems, driving post-training, reasoning, and evaluation to power enterprise AI deployments. Requires 7+ years ML research experience, PhD or equivalent, and strong publication record.

234k – 296k
Hybrid7+ YOEAI Research
The Voleon Group

Member of Research Staff, Reinforcement Learning

The Voleon GroupBerkeley, CA +1

Voleon Securities is seeking an experienced and creative reinforcement learning researcher to join their ML research group. This role focuses on applying state-of-the-art AI/ML techniques to construct market-making strategies and optimize trading models.

250k – 275k
HybridAI Research
Duolingo

Staff AI Research Engineer

DuolingoNew York, NY +1

Build and deploy AI systems (LLMs, bandits) for monetization, balancing revenue and retention. Requires advanced ML degree or equivalent, applied large-model experience, and leadership skills.

221k – 331k
On-site7+ YOEAI Research