AI Research Engineer: Scalable ML Systems & RL

Datadog, Inc.; Paris; Sur place; EUR 90 000 - 130 000; Plein temps

Avantages offerts par ce poste

Competitive benefits

RSUs + ESPP

Cross-office collaboration

Conference opportunities

Résumé du poste

Datadog in Paris is seeking a senior ML/ RL infrastructure engineer to design and operate large-scale training and inference systems, create multimodal data pipelines, and develop simulation environments for agent training.

You will work with Research Scientists, Product, and Engineering to turn prototypes into reliable services, optimize performance and cost, and contribute to publications and open-source artifacts.

Qualifications

• You have strong background in large-scale ML systems for training and inference.
• You know Ray, Slurm or similar frameworks and containerized deployments.
• You can explain design choices and performance trade-offs clearly.

Responsabilités

• 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
• Collaborate with Research Scientists, Product, and Engineering to integrate capabilities into Datadog's products and to harden prototypes into reliable services
• Contribute to research publications at top-tier conferences (e.g., NeurIPS, ICLR, ICML), and produce high-quality code, documentation, and open-source artifacts

Connaissances

Distributed computing

Reinforcement learning infra

ML systems at scale

Python

Rust/C++/Go

PyTorch/JAX

Megatron-LM/DeepSpeed

GPU acceleration

Outils

Ray

Slurm

Kubernetes

CUDA

Description du poste

Datadog in Paris is seeking a senior ML/ RL infrastructure engineer to design and operate large-scale training and inference systems, create multimodal data pipelines, and develop simulation environments for agent training.

You will work with Research Scientists, Product, and Engineering to turn prototypes into reliable services, optimize performance and cost, and contribute to publications and open-source artifacts.