AI Research Engineer: World Models & Scalable RL

Gomyneed Private Limited; Singapore; Hybrid; SGD 146,000 - 262,000; Full time

Job summary

Datadog is seeking an AI Research Engineer for the Datadog AI Research (DAIR) group, based in Paris, France. The role focuses on world models for observability and trained agents, collaborating with researchers to turn ideas into working systems.

You will build and operate large-scale training pipelines, evaluation infrastructure, and tooling, while advancing models and simulation environments for agent training and evaluation. Prior experience with distributed RL and ML systems is essential.

Qualifications

• Experience in distributed computing for training and inference at scale.
• Proficiency in Python and systems language (Rust/C++, Go).
• Hands-on with ML training/inference systems (PyTorch/JAX), containers, GPU acceleration.
• Experience with large-scale model training and fine-tuning (Megatron-LM, DeepSpeed, RLHF).
• Experience contributing to research publications.

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, agent performance, and simulation fidelity.

Skills

Distributed computing

RL infrastructure

Python programming

Systems languages

ML systems

Rust/C++/Go

Tools

PyTorch

JAX

Megatron-LM

DeepSpeed

Ray

Slurm

Job description

Datadog is seeking an AI Research Engineer for the Datadog AI Research (DAIR) group, based in Paris, France. The role focuses on world models for observability and trained agents, collaborating with researchers to turn ideas into working systems.

You will build and operate large-scale training pipelines, evaluation infrastructure, and tooling, while advancing models and simulation environments for agent training and evaluation. Prior experience with distributed RL and ML systems is essential.