Research Engineer develops and experiments with multimodal world models, focusing on data strategies, training techniques, evaluations, and production deployment for AI simulation technologies. Requires 4+ years in ML research/engineering and proficiency in PyTorch or JAX.
270k – 370k
Remote4+ YOEAI Research
About the role
Responsibilities
Run experiments to teach world models new behaviors — action following, scene manipulation, camera control, and beyond
Develop and test new data strategies, architectural variants, and training techniques
Design evaluations that measure model capabilities, and use them to drive model improvement
Take features from research prototype to production, collaborating with product and creative teams to address application-specific gaps
Contribute across the entire stack (data pipelines, modeling, production inference) to move projects forward
Requirements
4+ years of experience in machine learning research or engineering
Familiarity with the architecture, training, and inference of large-scale multimodal generative models
Experience building robust data pipelines for pretraining and post-training
Comfort working across the full research stack: data, training, evaluation, and deployment
Proficiency with at least one ML framework (e.g. PyTorch, JAX) and experience with distributed training at scale
Ability to context-switch quickly and drive projects forward in a fast-moving, ambiguous environment
Excitement about building AI that simulates the world
Lead a world-class research team advancing LLM scaling, post-training, RL, and inference efficiency at Databricks AI. Drive research roadmap and translate breakthroughs into production systems while collaborating closely with engineering and product teams.
270k – 340k
On-siteAI Research
Member of Technical Staff, Research Engineer (GPU Performance)
RunwayUnited States
Research Engineer optimizes GPU performance for large-scale AI world model training and real-time inference, focusing on custom kernels, distributed systems, and mixed-precision techniques. Requires 4+ years in ML infrastructure and performance optimization.
270k – 370k
Remote4+ YOEAI Research
Staff Machine Learning Research Scientist/ Engineer, Agents
Scale AISan Francisco, CA +2
Conducts cutting-edge research on data for state-of-the-art AI agents like browser and SWE agents, develops prototypes using LLMs and frameworks like PyTorch/JAX, and publishes in top ML venues. Requires 3+ years ML experience and strong cross-functional communication.
275k – 350k
On-siteAI Research
Staff Applied Scientist - Agentic Interfaces
DatadogNew York, NY
As a Staff Applied Scientist, you will define and build measurement systems for AI agent interfaces at Datadog, focusing on evaluation strategy, metric definition, and dataset creation to improve agent performance on customer workflows.
276k – 345k
On-site10+ YOEAI Research
Member of Technical Staff, Applied Research Scientist
RunwayUnited States
Applied Research Scientist develops computer vision and generative models for creative AI tools, collaborating cross-functionally to productionize research. Requires 4+ years in ML/CV/graphics, strong Python/C++/PyTorch skills, and experience mentoring researchers.