AI Research Scientist: World Models & Autonomous Agents

Datadog, Inc.; Pittsburgh (Allegheny County); On-site; USD 320,000 - 400,000; Full time

Benefits offered by this job

New hire stock equity (RSUs)

Employee stock purchase plan (ESPP)

Collaborate with Datadog offices in NY

Job summary

Datadog is seeking a Research Scientist to partner with Research Engineers and translate research advances into product capabilities in cloud observability and security.

You will work on world models, trained agents, and multimodal data from metrics, traces, logs, and events, collaborating with product and engineering teams to prototype, evaluate, and contribute to scalable AI-powered features across Datadog's platform.

Qualifications

• PhD in Computer Science, ML, or related field.
• Publications in top venues (NeurIPS/ICML/ICLR/TMLR) advantageous.

Responsibilities

• Conduct research in generative AI and foundation models for observability.
• Train multimodal models on large telemetry data using distributed training.
• Design simulated environments and RL loops for on-policy training and evaluation.
• Collaborate with Product and Engineering to integrate capabilities into Datadog products.
• Publish research findings and contribute to open-source model artifacts and benchmarks.

Skills

Generative modelling

World models

AI agents

Reinforcement learning

Multimodal learning

Distributed training

PyTorch

Education

PhD in Computer Science / ML / related field

Tools

DeepSpeed

Megatron-LM

PyTorch

Job description

Datadog is seeking a Research Scientist to partner with Research Engineers and translate research advances into product capabilities in cloud observability and security.

You will work on world models, trained agents, and multimodal data from metrics, traces, logs, and events, collaborating with product and engineering teams to prototype, evaluate, and contribute to scalable AI-powered features across Datadog's platform.