Staff AI Engineer - Notebooks

Datadog Datadog · Enterprise · Boston, MA +1 · Remote · Dev Eng

Staff AI Engineer focused on building AI-driven workflows within Datadog Notebooks, including agent systems, natural-language data analysis, and assisted graph generation. The role involves technical leadership, full development cycle ownership, and driving product direction for AI-powered features.

What you'd actually do

  1. Spearhead AI initiatives within Notebooks, serving as the technical owner for the design, development, and deployment of ML/AI-driven features.
  2. Shape the long-term technical direction and strategy for AI workflows within Notebooks and related products.
  3. Champion the full development cycle: from rapid prototyping and defining success metrics/evaluations to running experiments, iterating, and achieving production-ready scale and reliability.
  4. Drive product direction with a strong user focus, always advocating for the best end-user experience.
  5. Develop robust agent systems, including tools, retrieval and planning strategies, guardrails, prompt management/evaluations, fallbacks, and human-in-the-loop processes.

Skills

Required

  • Deep experience in AI/ML
  • 10+ years of relevant engineering experience
  • Experience acting as a technical lead
  • Proven track record of leading large-scale AI/ML initiatives
  • Significant experience in model deployment, development, training, fine-tuning, and evaluation

Nice to have

  • BS/MS/PhD in a scientific field or equivalent experience
  • Mentoring others
  • Cultivating a high-performing engineering culture
  • Leveraging AI tools to enhance how you code, solve problems, and build

What the JD emphasized

  • technical leader
  • product-driven environment
  • model deployment, development, training, fine-tuning, and evaluation

Other signals

  • AI-driven workflows
  • technical leader in AI
  • product-minded ML engineer
  • driving impactful initiatives end to end
  • building industry-leader AI-powered features