AI/ML Engineer

Title : AI/ML Engineer (All Levels)

Location : Bay Area (Hybrid)

Compensation : Up to $350,000 + Equity

We're partnered with a well-funded, high-growth Series B startup building the next generation of AI-native data infrastructure for Fortune 500 enterprises. With 3.5x revenue growth in FY'24 and 160% net revenue retention, they're scaling fast and investing heavily in the AI engineering team that powers their core product.

This is a high-impact role where you'll own the intelligence layer of the platform, making real architectural decisions around agent design, model selection, and evaluation strategy. Everything you build ships directly to customers; this is not a POC or internal tooling team.

What You'll Do

• Architect multi-step AI agents end-to-end, covering problem decomposition, context retrieval, error recovery, and output quality
• Own RAG pipelines, vector and graph-based retrieval systems, and knowledge representations that let agents reason accurately over complex enterprise data
• Build the AI layer behind the platform's core code generation capability, from prompt design and model selection through to fine-tuning and output validation
• Make production AI work at scale, balancing latency, throughput, and observability, and knowing when to optimize a model versus restructure a system
• Think in failure modes and evaluation strategy, not capability demos, and own the improvement loop

Requirements

• 4-6+ years of software engineering experience with meaningful AI/ML depth
• Proven experience building and shipping AI agents or coding agents in production
• Strong Python skills with hands-on experience across RAG, vector databases, and knowledge graphs
• Experience with LLM APIs, prompt engineering, fine-tuning, and structured outputs in production
• Familiarity with AWS and Kubernetes for production AI deployment
• Experience with data platforms such as Databricks, Snowflake, BigQuery, or Spark is a strong bonus
• Scale or Go experience is a bonus

If you're a builder who thinks in failure modes, owns what you ship, and thrives in environments where last year's best practice is already obsolete, this is a rare opportunity to have direct leverage on a product used by some of the world's largest enterprises.