AI Research Engineer - Datadog AI Research (DAIR)

About This Role

Key ResponsibilitiesAI-extracted

• 1
  Build and operate multimodal data pipelines, training and evaluation infrastructure, benchmarks, and internal tooling
• 2
  Implement models, run experiments at scale, and profile for reliability, performance, and cost
• 3
  Build simulation environments and replay infrastructure for agent training and evaluation
• 4
  Orchestrate distributed training and distributed RL with Ray, including scheduling, scaling, and failure recovery
• 5
  Collaborate with Research Scientists, Product, and Engineering to integrate capabilities into Datadog's products
• 6
  Contribute to research publications at top-tier conferences

RequirementsAI-extracted

• Depth in distributed computing, RL Infra, and ML systems for training and inference at scale
• Experience with Ray, Slurm, or similar frameworks is a plus
• Proficiency in Python, familiar with a systems language like Rust, C++, or Go
• Practical experience implementing and operating ML training and inference systems
• Experience with large-scale model training and fine-tuning, including frameworks like Megatron-LM, DeepSpeed, SkyRL, VeRL, or TorchTitan
• Ability to explain design and performance trade-offs clearly to both technical and non-technical audiences

Perks & BenefitsAI-extracted