AI Research Scientist - Datadog AI Research (DAIR)
$320k - $400k • New York, New York, USA; Pittsburgh, Pennsylvania, USA
Posted 18d ago
Job Location
New York, New York, USA; Pittsburgh, Pennsylvania, USA
Tech Stack
Remote Work Policy
On-site
Categories
AI Research Engineer
About the job
Datadog AI Research (DAIR) is seeking a Research Scientist to partner with Research Engineers on fundamental research problems. The role involves translating research advances into products by collaborating with Datadog's product and engineering teams. The team focuses on high-risk, high-reward problems in cloud observability and security, building on existing AI-powered solutions. The research areas include developing multimodal foundation models for understanding distributed systems and training autonomous agents for observability tasks like incident response and code repair.
Responsibilities
- 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 using distributed training infrastructure.
- Design and build simulated environments and RL training loops for agent training and evaluation.
- Collaborate with cross-functional teams 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.
- Contribute to research publications, present at top-tier conferences, and help open-source key model artifacts and benchmarks.
Requirements
- PhD in Computer Science, Machine Learning, or a related field, with deep expertise in generative modeling, world models, AI agents, reinforcement learning, or multimodal learning (or equivalent experience).
- Extensive experience designing and implementing deep learning models and agents.
- Strong background in distributed training frameworks (e.g., DeepSpeed, Megatron-LM) and ML libraries (PyTorch).
- Track record of impactful publications at top-tier venues (e.g., NeurIPS, ICLR, ICML, TMLR).
- Familiarity with efficient training, post-training, and inference techniques for large foundation models.
- Ability to explain complex models and research findings to both technical and non-technical audiences.
Benefits
- Competitive global benefits
- New hire stock equity (RSUs) and employee stock purchase plan (ESPP)
- Opportunity to collaborate closely with colleagues across Datadog offices in New York City and Paris
- Opportunity to attend and present at conferences and meetups
- Intra-departmental mentor and buddy program for in-house networking
- Inclusive company culture, ability to join Community Guilds (Datadog employee resource groups)