AI Research Engineer - Scalable ML Systems & Autonomous Agents

Rozgoo; New York (NY); On-site; USD 170,000 - 230,000; Full time

Job summary

Datadog is seeking a Research Engineer to partner with researchers and turn ideas into scalable, production-ready systems. You will help build data pipelines, training and evaluation infrastructure, and tooling to support rapid experimentation.

You will work on world models for observability and autonomous agents, contribute to benchmarks, and collaborate with product and engineering teams to harden prototypes into reliable services.

Qualifications

• Deep expertise in distributed computing and ML systems.

Responsibilities

• Build and operate multimodal data pipelines, training and evaluation infrastructure.

Skills

Distributed computing

RL Infra

ML systems

Python

Rust

C++

Go

PyTorch

JAX

containerization

orchestration

GPU acceleration

Megatron-LM

DeepSpeed

SkyRL

VeRL

TorchTitan

SFT

RLVR

RLHF

Tools

Ray

Slurm

Megatron-LM

DeepSpeed

SkyRL

VeRL

TorchTitan

PyTorch

JAX

Job description

Datadog is seeking a Research Engineer to partner with researchers and turn ideas into scalable, production-ready systems. You will help build data pipelines, training and evaluation infrastructure, and tooling to support rapid experimentation.

You will work on world models for observability and autonomous agents, contribute to benchmarks, and collaborate with product and engineering teams to harden prototypes into reliable services.