Staff AI Engineer - Notebooks

About the Role

The role entails working closely with cross-functional teams to design, develop, and deploy ML/AI-driven features, as well as shaping the long-term technical direction and strategy for AI workflows within Notebooks and related products. You will be expected to champion the full development cycle, from rapid prototyping to production-ready scale and reliability, and drive product direction with a strong user focus.

What You Will Do

• Spearhead AI initiatives within Notebooks, serving as the technical owner for the design, development, and deployment of ML/AI-driven features
• Shape the long-term technical direction and strategy for AI workflows within Notebooks and related products
• Champion the full development cycle: from rapid prototyping and defining success metrics/evaluations to running experiments, iterating, and achieving production-ready scale and reliability
• Drive product direction with a strong user focus, always advocating for the best end-user experience
• Coordinate with other leaders and product teams interested in leveraging Notebooks in their AI-powered workflows
• Develop robust agent systems, including tools, retrieval and planning strategies, guardrails, prompt management/evaluations, fallbacks, and human-in-the-loop processes
• Stay up to date with industry trends, guiding the adoption of new technologies and evolving best-practices
• Collaborate with the engineering team to develop and refine AI workflows and features
• Participate in code reviews and ensure high-quality code is being delivered

What We Are Looking For

• A BS/MS/PhD in a scientific field or equivalent experience
• 10+ years of relevant engineering experience, as well as experience acting as a technical lead
• Proven track record of leading large-scale AI/ML initiatives in a product-driven environment
• Significant experience in model deployment, development, training, fine-tuning, and evaluation
• Passion for mentoring others and cultivating a high-performing engineering culture
• Experience with Excel and proficiency in using AI coding tools in day-to-day workflows
• Demonstrated ability to use AI coding tools to enhance how you code, solve problems, and build
• Motivation to push the boundaries of how AI can improve software engineering best practices and contribute to building AI-enabled products

Nice to Have

• Experience with cloud-based AI platforms and tools
• Knowledge of natural language processing and computer vision
• Familiarity with agile development methodologies
• Experience working in a remote or distributed team environment

Benefits and Perks

• Generous and competitive benefits package
• New hire stock equity (RSUs) and employee stock purchase plan
• Continuous career development and pathing opportunities
• Employee-focused best-in-class onboarding
• Internal mentor and cross-departmental buddy program
• Friendly and inclusive workplace culture
• Remote work stipend and flexible work arrangements
• Access to cutting-edge technologies and tools
• Opportunities for professional growth and advancement

• To stand out as a candidate, highlight your experience with AI-powered features and ML engineering, and be prepared to provide examples of your work.
• Make sure to showcase your technical leadership skills, including your ability to drive impactful initiatives end-to-end and cultivate a high-performing engineering culture.
• Familiarize yourself with Datadog's products and services, and be prepared to discuss how you can contribute to the company's mission and goals.
• Be prepared to discuss your experience with Excel and AI coding tools, and how you have used these tools to enhance your work.
• Consider creating a portfolio of your work, including examples of your AI-powered features and ML engineering projects, to demonstrate your skills and experience to potential employers.
• When negotiating salary, be sure to research the market rate for similar positions and be prepared to make a strong case for your worth as a candidate.