Senior AI Research Engineer - LLM Training & Inference Infrastructure

Job description preview

From Datadog’s original posting

As a Research Engineer on the Frontier Foundations team, you will partner with Research Scientists to build the systems that run large-scale LLM training, reinforcement learning, and inference at Datadog. You will build and operate an in-house distributed runtime that schedules work across large GPU clusters, and you will own how quickly researchers can iterate and how reliably long training runs complete. Building on our track record of AI-powered solutions (e.g., Bits AI, Bits Evolve, and our time series foundation model Toto), Datadog AI Research (DAIR) tackles high-risk, high-reward problems grounded in real-world challenges in cloud observability and security. The Frontier Foundations team builds and maintains the high-performance infrastructure that lets us train and serve large language models at scale. It supports both of our research areas: Trained Agents for Observability -- Post-training models to operate autonomously across Datadog's domain. SRE incident response is our first target, with a clear path to code repair, security response, and infrastructure optimization. We build the simulation environments, RL training loops, and evaluation infrastructure needed to train…