AI Research Scientist - Datadog AI Research (DAIR)

What you'll do

• Research and train foundation models and agents using large-scale telemetry data and distributed training infrastructure.
• Build simulated environments, reinforcement-learning training loops, and evaluation infrastructure for agents.
• Collaborate with Product and Engineering to integrate research into products, and contribute to publications and open-source artifacts.

What they ask for

• PhD in Computer Science, Machine Learning, or a related field, or equivalent experience.
• Deep expertise in generative modeling, world models, AI agents, reinforcement learning, or multimodal learning.
• Extensive experience designing and implementing deep learning models and agents, including distributed training frameworks and ML libraries.
• Track record of impactful publications at top-tier venues; familiarity with efficient training, post-training, and inference for large foundation models.

Nice to have

• Experience applying research to products, especially large foundation models, world models, or RL-trained agents.
• Experience with production data pipelines or applications, or GPU programming and optimization including CUDA.

Pay

$320,000 — $400,000 USD (yearly salary; base or total pay not specified)

Tools

DeepSpeed, Megatron-LM, PyTorch, CUDA, NeurIPS, ICLR, ICML, TMLR

AI focus

Develop multimodal world models and RL-trained autonomous agents for observability, including forecasting, anomaly detection, root cause analysis, simulation, and incident response.