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)
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
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
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
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
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.