Staff AI Engineer
About The Position
We're looking for a Staff AI Engineer to help advance Amplitude's AI products and the foundations they're built on. As AI reshapes how customers analyze data, build products, and act on insights, we need engineers who can move fluidly between deep technical work and product judgment, shipping AI capabilities that customers and internal teams actually rely on. This role is both visionary and hands-on. You'll identify where AI creates the most leverage, build the systems that deliver it, and establish the patterns that let the rest of the company build faster on top of your work. You'll contribute to customer-facing AI products, the platform infrastructure beneath them, and the internal tooling that helps Amplitude operate as an AI-native company.
Requirements
- 5+ years of software engineering experience, with 2+ years building production LLM or applied AI systems
- Experience designing and scaling AI products, platforms, or developer tools with real users and real reliability constraints
- Deep familiarity with the modern AI ecosystem: model APIs, agent frameworks, evals, retrieval, prompting, tool use, and the tradeoffs between them
- Strong product sense and systems thinking; you know when to prototype, when to harden, and when to invest in durable infrastructure
- Excellent communicator and collaborator, able to drive alignment across engineering, design, product, and go-to-market
Nice To Haves
- Highest-leverage problem identification: You consistently spot the highest-value opportunities and use data to focus effort where it matters most
- Systems thinking: You see across disparate tools, data sources, and teams, and connect them into cohesive, contextual AI-powered experiences
- Hands-on design and implementation of agentic systems: You design and ship robust agentic workflows with strong context handling, orchestration, tool use, and evaluation
- Build vs. buy judgment: You know when to adopt existing solutions and when to build from scratch
- Driving adoption & evangelism: You get capabilities into users' hands, onboard them effectively, and evolve based on feedback
- Product thinking: You blend technical execution with product intuition to build AI experiences that people genuinely love to use
Responsibilities
- Ship high-impact AI features and products that change how customers and internal teams work, from agentic workflows to embedded intelligence across the Amplitude platform
- Design and build the foundational infrastructure behind Amplitude's AI systems: orchestration, evaluation, retrieval, context management, and integrations with the broader product ecosystem
- Identify high-leverage opportunities by deeply understanding user workflows, customer pain points, and emerging capabilities in the AI ecosystem
- Establish reusable patterns, primitives, and components that enable other engineers to build AI features confidently and consistently
- Drive adoption and quality through close collaboration with product, design, and customer-facing teams, with a relentless focus on real usage and measurable outcomes
- Set technical direction for AI systems at Amplitude, making thoughtful build vs. buy calls, shaping evaluation strategy, and influencing the long-term roadmap
Benefits
- Excellent Medical, Dental and Vision insurance coverages, with 100% employer-paid premiums for employee Medical, Dental, Vision on select plans
- 401(k) retirement plan with an employer match of up to 1% of your eligible pay each pay period up to $2,000 annually
- Flexible time off, paid holidays, and more
- Generous stipends to spend on what matters most to you, whether that’s wellness (monthly), commuter transit/parking (monthly), learning and development (quarterly), new hire home office equipment, and much more
- Excellent Parental benefits including: 12 weeks of Paid Parental Leave, Carrot Fertility Benefits/Adoption/Surrogacy support, Back-up Child Care support
- Mental health and wellness benefits including no cost employee access to Modern Health coaching & therapy sessions
- Employee Stock Purchase Program (ESPP)
- Equity, benefits and other forms of compensation
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
Education Level
No Education Listed
