AI Research Engineer - Multimodal World Models & Agents

Academic Key; Paris; On-site; EUR 90,000 - 150,000; Full time

Benefits offered by this job

Stock equity (RSUs)

ESPP

Conference participation

Mentor program

In-house networking

Job summary

Datadog is seeking an AI Research Engineer to collaborate with Research Scientists, building data, tooling, and infrastructure to transform research ideas into scalable, production-ready systems for cloud observability and security.

You will help advance world models for observability and trained agents, training at scale with modern RL and ML frameworks, and contribute to publications while partnering with Product and Engineering to harden prototypes into reliable services.

Qualifications

• Depth in distributed computing, RL infra, and ML systems for training and inference at scale.
• Experience with Ray, Slurm, or similar frameworks.
• Proficiency in Python and systems languages; familiarity with cloud and data infra.
• Experience with ML training/inference systems (PyTorch/JAX), containerization, orchestration, and GPU acceleration.
• Hands-on experience with large-scale model training and fine-tuning (Megatron-LM, DeepSpeed, SkyRL, VeRL, TorchTitan).

Responsibilities

• Build and operate multimodal data pipelines, training and evaluation infrastructure, benchmarks, and internal tooling.
• Implement models, run experiments at scale, and profile for reliability, performance, and cost.
• Build simulation environments and replay infrastructure for agent training and evaluation.
• Orchestrate distributed training and distributed RL with Ray, including scheduling, scaling, and failure recovery.
• Establish rigorous automated benchmarks and regression tests for world model predictions and agent performance.
• Collaborate with Research Scientists, Product, and Engineering to integrate capabilities into Datadog's products.
• Contribute to research publications and produce high-quality code, docs, and open-source artifacts.

Skills

Distributed computing

RL infrastructure

ML systems

Python

Systems programming

Tools

Ray

Slurm

Megatron-LM

DeepSpeed

JAX

PyTorch

CUDA

Job description

Datadog is seeking an AI Research Engineer to collaborate with Research Scientists, building data, tooling, and infrastructure to transform research ideas into scalable, production-ready systems for cloud observability and security.

You will help advance world models for observability and trained agents, training at scale with modern RL and ML frameworks, and contribute to publications while partnering with Product and Engineering to harden prototypes into reliable services.