Staff AI Engineer - Notebooks | Datadog Careers

Datadog

Hybrid

Quick summary

Work type
Hybrid
Location
Salary
$234,000–$300,000 / yr
Posted
15 days ago

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $208k
This role $267k
$152k most similar roles pay here $316k

This role pays more than 88% of similar roles. Most pay $170,000–$246,150 — the shaded band above. At the midpoint, this role pays about $267k versus about $208k for comparable roles.

Based on 240 similar postings.

Employer

About Datadog

Datadog, Inc. is an American company that provides an observability service for cloud-scale applications, providing monitoring of servers, databases, tools, and services, through a SaaS-based data analytics platform.

Datadog currently has 109 open roles on FindRole.

Listed pay typically runs $187,000–$240,000 across 58 roles with salary data.

Most-posted roles

View all roles at Datadog

At a glance

TL;DR · Staff AI Engineer - Notebooks | Datadog Careers

As a Staff Software Engineer on the Notebooks team at Datadog, you will lead AI-driven initiatives by designing and deploying machine learning features that enhance data analysis and collaboration. Your role involves setting long-term technical direction for AI workflows, overseeing the full development cycle from prototyping to production readiness, and ensuring robust agent systems are in place. You will work closely with other product teams to integrate Notebooks into their AI-powered solutions while maintaining a strong focus on user experience. Ideal candidates have over 10 years of engineering experience, including technical leadership roles, with expertise in model deployment, training, and evaluation. Proficiency in Python, familiarity with cloud platforms like AWS or GCP, and knowledge of machine learning frameworks such as TensorFlow or PyTorch are essential. This role demands a deep understanding of AI trends and the ability to mentor others while fostering a high-performing engineering culture.

What you'll do

  • Spearhead the design, development, and deployment of ML/AI-driven features in Notebooks.
  • Shape long-term technical direction for AI workflows within Notebooks and related products.
  • Drive full development cycle from prototyping to production-ready scale and reliability.
  • Coordinate with other leaders and teams on leveraging Notebooks for AI-powered workflows.
  • Develop robust agent systems including tools, retrieval strategies, guardrails, and human-in-the-loop processes.

What we're looking for

  • 10+ years of relevant engineering experience with a focus on AI/ML.
  • Proven leadership in large-scale AI/ML initiatives within product-driven environments.
  • Strong expertise in deploying, developing, training, fine-tuning, and evaluating ML models.
  • Ability to shape long-term technical direction and drive impactful AI workflows.
  • Excellent communication skills for coordinating across teams and driving product direction.
  • Passionate about mentoring and fostering a high-performing engineering culture.

More like this

Similar roles

Senior Staff Engineer, AI Software

Samsung Semiconductor

San Jose, CA 5 days ago $189,000$301,000
PyTorch vLLM Linux GitHub Jira High Bandwidth Memory (HBM) LLM model architectures transformer-based designs agentic AI architecture memory-centric acceleration development tooling Git CI/CD

Staff Engineer - Applied AI

GEICO

Remote (Bethesda, MD) 77 days ago $115,000$230,000
Python Java LLMs Kubernetes Docker CI/CD Prometheus Grafana PostgreSQL Redis Knowledge_Graphs Vector_Databases Spring_AI LangChain4j Embabel AWS Azure Git Jenkins GitHub Swagger RESTful_APIs
Remote

Staff Software Engineer - AI Applications

Plaid

San Francisco, CA 80 days ago $228,360$337,260
LLMs Prompt engineering Fine-tuning RAG Vector database Embedding models Agent orchestration framework Evaluation and monitoring framework Streaming SSE HTML CSS JavaScript Python Docker CI/CD Kubernetes PostgreSQL AWS GitHub MVP

Careers

Qualcomm

Austin, TX 104 days ago $134,800$202,200
Python LangChain LangGraph AsyncIO Pydantic LLMs Embedding Models Vector Stores Distributed Systems Micorservices REST Docker Kubernetes CI/CD Milvus Qdrant Chroma Neo4j Ragas DeepEval