Staff AI Engineer - Notebooks

Requirements:

• You have 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
• You have significant experience in model deployment, development, training, fine-tuning, and evaluation
• You are passionate about mentoring others and cultivating a high-performing engineering culture
• You’re excited about leveraging AI tools to enhance how you code, solve problems, and build

What You'll Be Doing:

• 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

Perks and Benefits:

• 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