AI Research Scientist — Digital World Models
About the role
We are looking for exceptional AI Research Scientists to advance the state of the art in Digital World Models.
You will work on fundamental research problems at the intersection of world models, representation learning, self-supervised learning, sequential prediction and large-scale machine learning.
Unlike models of the physical world, Digital World Models must learn from heterogeneous, highly structured and partially observed systems composed of resources, events, metrics, dependencies and actions. This creates a new class of research problems around representation, dynamics, causality and intervention.
You will help define those problems — and build the models that solve them.
What you will do
- Conduct original research on Digital World Models, developing new architectures for state representation, temporal modeling, latent dynamics and action-conditioned prediction.
- Develop models that capture interactions between resources, workloads, events, dependencies and autonomous agents across multiple temporal and structural scales.
- Study prediction under intervention and counterfactual futures — moving from what happens next? to what happens if we do X?
- Work closely with Research Engineers to scale datasets, training and evaluation, turning promising ideas into large-scale experiments.
- Help define PRESAGE's scientific roadmap and contribute to significant publications and research releases.
Who you are
- You hold a PhD in Computer Science, Artificial Intelligence, Machine Learning or a closely related field.
- You have a strong research track record in machine learning and deep expertise in one or more areas relevant to world modeling, such as representation learning, generative or predictive modeling, sequential modeling, reinforcement learning, multimodal learning or foundation models.
- You can turn open-ended scientific questions into clear hypotheses, experiments and measurable results.
- You have strong practical deep-learning experience and can independently implement and evaluate new ideas using PyTorch.
- You care deeply about experimental rigor, reproducibility and understanding why a model works — not only whether a metric improves.
- You are excited by research environments where important questions and architectures are still being invented.
Strong additions
- Publications at leading ML or AI conferences.
- Experience training large models or working with distributed training infrastructure.
- Experience with latent world models, predictive representation learning or model-based agents.
- Experience modeling structured, relational, temporal or event-based data.
- Familiarity with structured temporal data, distributed systems or cloud infrastructure.
- Contributions to widely used open-source research projects or significant model releases.