# AI Research Engineer – Datadog AI Research (DAIR)
**Company:** [Datadog](https://hotfix.jobs/companies/datadog)
**Location:** New York, NY
**Salary:** $140K-$400K
**Skills:** Python, PyTorch, JAX, Ray, Rust, C++, Go, Kubernetes, CUDA, Slurm
**Posted:** 2026-01-21
> Builds ML infrastructure and tooling to productionize AI research in observability models, SRE agents, and code repair. Requires strong Python/ML systems expertise, distributed computing experience, and proficiency in PyTorch/JAX.
## Job Description
## What You’ll Do:
- Build and operate datasets, training and evaluation pipelines, benchmarks, and internal tooling
- Implement models, run experiments at scale, and profile for reliability, performance, and cost
- Orchestrate distributed training and distributed RL with **Ray**, including scheduling, scaling, and failure recovery
- Make the research stack observable, reproducible, and easier to use
- Establish rigorous automated benchmarks and regression tests for forecasting, anomaly detection, multi-modal analysis, agents, and code repair tasks
- Collaborate with Research Scientists, Product, and Engineering to integrate advanced AI capabilities into Datadog’s product ecosystem
- Contribute high-quality code, documentation, and open-source artifacts

## Who You Are:
- Strong software engineering skills with experience in observability, SRE, or security
- Depth in distributed computing and ML systems; experience with **Ray**, **Slurm**, or similar
- Proficient in **Python**, familiar with systems languages (**Rust**, **C++**, **Go**), comfortable with cloud and data infrastructure
- Practical experience with **PyTorch** or **JAX**, containerization, orchestration, and GPU acceleration
- Familiar with efficient training, fine-tuning, and inference for large foundation models

## Bonus Points:
- Experience bridging research prototypes to products with foundation models or AI agents
- Hands-on with GPU programming (**CUDA**)
- Experience with production data pipelines

## Benefits and Growth:
- Competitive global benefits
- New hire stock equity (RSUs) and ESPP
- Opportunities to collaborate across NYC and Paris offices
**Apply:** https://hotfix.jobs/jobs/ai-research-engineer-datadog-ai-research-dair-at-datadog-fa816d65-77b3-445d-a8d8-536e89461a90
**Canonical:** https://hotfix.jobs/jobs/ai-research-engineer-datadog-ai-research-dair-at-datadog-fa816d65-77b3-445d-a8d8-536e89461a90