Staff AI Engineer - Notebooks
WFA Digital Insight
The demand for AI engineers with expertise in machine learning has surged in recent years, with over 50% of companies adopting AI technologies. As a Staff AI Engineer at Datadog, you'll be at the forefront of innovation, driving AI initiatives and shaping the long-term technical direction for AI workflows. With the rise of remote work, companies are now looking for talent with a strong technical lead background and experience in product-driven environments. Datadog, a global SaaS business, offers a unique opportunity to work on industry-leading AI-powered features and collaborate with a team of experts.
Job Description
About the Role
The Staff AI Engineer role at Datadog is a unique opportunity for experienced engineers to lead AI initiatives and drive product direction. As a technical leader, you will be responsible for building industry-leading AI-powered features and collaborating with a team of experts to deliver high-quality products. The Notebooks team is focused on developing AI-driven workflows for natural-language data analysis, assisted graph generation, automated investigations, and more.The role entails working closely with the product team to shape the long-term technical direction and strategy for AI workflows within Notebooks and related products. You will act as a technical owner for the design, development, and deployment of ML/AI-driven features, ensuring that they meet the highest standards of quality and reliability.
The Notebooks team is part of Datadog's larger engineering organization, which is committed to delivering innovative solutions to complex problems. As a Staff AI Engineer, you will have the opportunity to work with a talented team of engineers, product managers, and designers to drive the development of AI-powered products.
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 identify and prioritize areas for improvement in the AI workflows
- Develop and maintain technical documentation for AI workflows and related products
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
- Strong technical expertise in AI/ML, with a focus on natural-language processing and machine learning
- Experience with Excel and other data analysis tools
- Excellent communication skills, with the ability to communicate complex technical concepts to non-technical stakeholders
Nice to Have
- Experience with cloud-based technologies, such as AWS or Google Cloud
- Knowledge of agile development methodologies and version control systems, such as Git
- Familiarity with DevOps practices and tools, such as Docker and Kubernetes
- Experience with data visualization tools, such as Tableau or Power BI
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
- Flexible working hours and remote work options
- Access to cutting-edge technologies and tools
- Opportunities for professional growth and advancement
How to Stand Out
- Develop a strong portfolio of AI/ML projects, including experience with natural-language processing and machine learning
- Showcase your ability to communicate complex technical concepts to non-technical stakeholders
- Highlight your experience with cloud-based technologies, such as AWS or Google Cloud
- Be prepared to discuss your experience with agile development methodologies and version control systems, such as Git
- Research Datadog's products and services, and be prepared to discuss how your skills and experience align with the company's goals and objectives
- Practice your coding skills, including experience with languages such as Python and Java
- Prepare to discuss your experience with data visualization tools, such as Tableau or Power BI
This is a remote position listed on WFA Digital, the platform for professionals who work from anywhere. Browse more remote jobs across all categories.