Staff AI Platform Engineer

📋 Description

• Design and operate the knowledge graph and ontology for grocery data relationships.
• Build retrieval systems (vector, graph, structured) to feed reliable model context.
• Develop LLM-powered agents and the serving infra to run them cost-effectively.
• Create tools and interfaces for engineers — a platform, not one-off features.
• Own evaluation, quality, and the data foundation; define metrics and guardrails.

🎯 Requirements

• 5+ years building production software, data, or ML systems; strong Python API design.
• Hands-on LLM systems: retrieval/RAG, agents, tool-use, prompt/context engineering, evaluation.
• Data engineering and platforms: pipelines, modeling, Databricks/Spark, MLflow, cloud warehouses.
• Comfort with AI system trade-offs: retrieval quality, latency, cost; build guardrails and evals.
• Platform mindset: leverage with clean interfaces; thrive in ambiguity.

🎁 Benefits

• Health, dental, vision coverage with premiums covered.
• Competitive salary, equity (US), and 401(k) with company match.
• Flexible workspace: home office stipend and coworking wallets.
• Growth-focused culture; annual professional development budget.
• Monthly stipends for Betterment and telecom.
• Flexible paid time off.