Staff & Principal AI/ML Engineer

About the role

We're partnering with a well-funded, high-growth Series B startup that builds the next generation of AI-native data infrastructure for Fortune 500 enterprises. This role is for someone who sets technical direction, not just executes it.

Responsibilities

• Set technical direction across multi-step agent architecture, RAG pipelines, and knowledge retrieval systems, making the calls on agent design, model selection, and build vs. buy
• Own the AI behind the platform's core code generation capability, from prompt design and fine-tuning through to output validation at scale
• Lead large, complex initiatives end-to-end with minimal direction, defining the problem as much as the solution
• Shape how production AI gets built across the team, establishing standards for evaluation, observability, and reliability
• Influence architectural decisions across the broader engineering organization and mentor engineers at all levels

Requirements

• 8+ years of software engineering experience with significant AI/ML depth and a track record of technical leadership
• Demonstrated experience owning and shipping AI agents or coding agents in production at scale
• Strong Python skills with deep hands-on experience across RAG, vector databases, and knowledge graphs (Neo4j)
• Ability to make architectural decisions under uncertainty and drive them through to production
• Experience with AWS and Kubernetes, and comfort operating across the full AI development lifecycle
• Experience with data platforms such as Databricks, Snowflake, BigQuery, or Spark is a strong plus
• Scala or Go experience is a bonus