Deep Learning Research Engineer Intern

Data Robot delivers AI that maximizes impact and minimizes business risk. Our platform and applications integrate into core business processes so teams can develop, deliver, and govern AI aRobot empowers practitioners to deliver predictive and generative AI, and enables leaders to secure their AI assets. Organizations worldwide rely on Data Robot for AI that makes sense for their business — today and in the future.

This effort is about building a multi‑target joint probabilistic foundation model that can be used across industries to tackle some of the hardest real‑world problems. The ambition goes beyond what tabular and time‑series foundation models usually do: one model should support temporal forecasting, unordered tabular regression and classification, missing‑data completion, and mixed‑modality inputs and outputs while learning coherent joint structure across connected variables, rows, horizons, and scenarios.

The primary objective is to create real value in serious use cases rather than optimize benchmark scores in isolation, although strong results on standard benchmarks are expected to follow closely. This role is designed as a research‑engineering internship for someone operating at post‑doctoral level, or very close to it, who wants to shape the architecture of a probabilistic foundation model.

The center of gravity is the model itself: the encoders that read temporal, tabular, and mixed‑modality inputs, the attention and sequence mechanisms that carry structure across variables, rows, and horizons, and the distributional output heads and decoding schemes that turn hidden states into coherent joint predictions. We are looking for someone who can reason precisely about what an architecture can and cannot represent, and who then implements, trains, and evaluates the resulting PyTorch models in a production‑quality codebase.

A background in stochastic modeling is a strong plus rather than a prerequisite: the model is trained on synthetic data drawn from stochastic dynamics, so a candidate who can also reason about stochastic differential equations and design synthetic data generators can contribute on both sides of the training loop.

Strong PyTorch skills and hands‑on experience building and training deep models. Solid understanding of Transformers, attention variants, and long‑context or state‑space sequence architectures, including the tradeoffs between quality, memory, and latency. Experience with probabilistic modeling in neural networks: distributional output heads, likelihood‑based or proper‑scoring‑rule losses, mixture or flow models, or related uncertainty‑aware learning setups. Strong foundation in probability and statistics, enough to reason about densities and masses, dependence between variables, and calibration.

Ability to connect architectural ideas to working GPU‑native implementations, controlled experiments, and diagnostics that show why a change helped. Strong engineering habits: readable code, tests, reproducible experiments, and disciplined evaluation of model changes. Ability to debug training instability, reason about why results changed, and iterate quickly from hypothesis to evidence. Strong Plus:
Working knowledge of stochastic processes and stochastic differential equations: drift and diffusion, jumps, regime switching, mean reversion, heavy tails, and how such dynamics are simulated numerically. Experience designing synthetic data generators or simulation‑based training curricula, and an understanding of how the training distribution shapes what a model learns. Depth in a domain with rich stochastic structure such as finance, energy, commodities, or a similarly quantitative field.

Familiarity with tabular or…