AI Research Scientist - Datadog AI Research (DAIR)

Job summary

Responsibilities

As a Research Scientist on our team, you will partner with Research Engineers, working on fundamental research problems and collaborating with Datadog's product and engineering teams to translate research advances into products.

• Conduct research in generative AI and machine learning, building specialized foundation models and trained agents for observability
• Train multimodal models on large-scale, diverse telemetry data (metrics, logs, traces, topology, events) using distributed training infrastructure
• Design and build simulated environments and RL training loops for on-policy agent training and evaluation
• Collaborate with cross-functional teams (Product, Engineering) to integrate capabilities like multimodal world modeling and autonomous agents into Datadog's products
• Stay at the forefront of foundation models, world models, and RL-based agent research

What they're looking for

• You hold a PhD in Computer Science, Machine Learning, or a related field, with deep expertise in areas like generative modeling, world models, AI agents, reinforcement learning…
• You have extensive experience designing and implementing deep learning models and agents, with a strong background in distributed training frameworks (e.g., DeepSpeed…
• You have a track record of impactful publications at top-tier venues (e.g., NeurIPS, ICLR, ICML, TMLR)
• You are familiar with efficient training, post-training, and inference techniques for large foundation models
• You can explain complex models and research findings to both technical and non-technical audiences

Tech stack

• PyTorch
• CUDA