AI Research Engineer - Datadog AI Research (DAIR)
Core
Build data pipelines, training/evaluation infrastructure, and simulation environments to turn research ideas into working systems for AI-powered observability and security.
Role type
Senior IC AI Research Engineer (RL Infra & Systems)
Builds
Multimodal foundation models, trained autonomous agents, simulation environments, and evaluation infrastructure for cloud observability.
Domain
Cloud Observability, Security, AI/ML Systems
Deliverable
production ML models | infrastructure | research
Required skills
Distributed computing, RL Infra, ML systems for scale, Python, Systems languages (Rust/C++/Go), Cloud/data infrastructure, ML training/inference ops, Large-scale model training/fine-tuning, Automated benchmarking
Preferred skills
Observability/SRE/Security domain knowledge, Bridging prototypes to products, GPU programming (CUDA), Production data pipelines, Simulation/sandbox environments
Technologies
Ray, Slurm, PyTorch, JAX, Megatron-LM, DeepSpeed, SkyRL, VeRL, TorchTitan, CUDA
Responsibilities
Build multimodal data pipelines and training/evaluation infrastructure; Implement models and run large-scale experiments; Build simulation environments for agent training; Orchestrate distributed training and RL; Establish automated benchmarks; Collaborate on product integration and research publications.
Seniority
Senior, hands-on IC