Staff AI Platform Engineer

Afresh, the AI platform for grocery, began by tackling the most complex problem in the industry: fresh, and has evolved into the core AI platform for grocers.

By leveraging proprietary AI designed for high-volatility environments, we empower partners like Albertsons, Meijer, and Wakefern to drive smarter decisions across their entire enterprise.

Following record-breaking 70% revenue growth in 2025, we have scaled to 6 enterprise-grade solutions, with solutions live in over 10% of the U.S. grocery market. Our platform now orchestrates billions of decisions from the store floor to the distribution center and prevented over 200 million pounds of food waste last year alone.

If you're looking for a role where your work directly translates into massive scale and social good, and you want to be part of the team that defines how the world eats, there is no better time to join us.

About the Role

Frontier models are a commodity. The knowledge you feed them is not.

As a Staff AI Platform Engineer, you build the AI and data platform that powers Afresh's products: the knowledge and retrieval layer that makes grocery data reliably usable by LLMs, the agent systems built on top of it, and the evaluation and serving infrastructure underneath. Your "customers" are Afresh's own engineers and AI products — your job is to give them a platform that turns raw grocery data into context a model can be trusted with, at production quality.

This is senior, 0-to-1 platform work. You'll make foundational choices about how we represent grocery knowledge, ground our models, and measure whether any of it is actually working — in a fast-moving space with no playbook.

What You’ll Do

Build the knowledge & retrieval layer

• Design and operate the knowledge graph and ontology that capture how grocery data relates.
• Build the retrieval systems (vector, graph, and structured) that feed the right context to our models — so grounding is reliable, not lucky.

Build and serve the agent platform

• Build LLM-powered agents (tool-use, multi-step reasoning, orchestration) and the serving infrastructure to run them reliably and cost-effectively.
• Build the tools, abstractions, and interfaces other engineers depend on — a platform, not one-off features.

Own evaluation, quality, and the data foundation

• Stand up eval sets, LLM-as-judge harnesses, tracing, and observability, plus the metrics (faithfulness, accuracy, hallucination rate, latency, cost) that tell us whether a change helped or hurt.
• Build the pipelines, data products, and experimentation on Databricks/MLflow that take work from prototype to production — and partner with the engineers deploying our AI in the field to harden what works into reusable capabilities.

What Makes You a Good Fit

We encourage all highly-qualified candidates to apply, even if they do not fulfill all the listed criteria.

• 5+ years building production software, data, or ML systems; an excellent engineer with strong systems and API design (Python).
• Hands-on production experience with LLM systems: retrieval/RAG, agents and tool-use, prompt and context engineering — and, critically, evaluation. You measure quality; you don't eyeball it.
• Solid data-engineering and data-platform foundations: pipelines, data modeling, and a modern cloud data stack (Databricks/Spark, MLflow, cloud warehouses).
• Comfort in the messy middle of AI systems — retrieval quality, latency and cost trade-offs, non-determinism — and the instinct to build the guardrails and evals that make them trustworthy.
• A platform mindset: you build for leverage and clean interfaces, and you thrive in ambiguity in a fast-moving space.

Nice to Have

• Knowledge graphs, ontologies, or semantic layers in production; graph databases.
• Vector stores (pgvector, Pinecone, Weaviate, etc.) and hybrid search.
• MCP or similar tool/context protocols; agent frameworks (e.g., LangGraph).
• MLOps and model serving at scale; experimentation and observability tooling for LLM systems.
• Experience in grocery, retail, or other complex enterprise data domains.

Our Tech Stack

• Python, PySpark, dbt
• Databricks (Delta Lake, Unity Catalog, MLflow)
• Astronomer (Airflow) for orchestration
• LLMs and agents (Claude), retrieval over vector + graph stores, MCP tooling, eval and observability harnesses
• Claude, GitHub, Shortcut, Notion for development workflows

Why Afresh?

• We're a mission-driven company that eliminates hundreds of millions of pounds of food waste in grocery stores every year — your platform powers products with direct, visible impact.
• Build the foundation the rest of the product stands on: high-leverage, 0-to-1 work where your abstractions make the whole team faster.
• Be part of an engineering culture that's genuinely AI-forward — we want to be on the bleeding edge of agentic systems, not watching from the sidelines.
• Senior team, high trust, and real ownership at a pivotal inflection point for how Afresh scales.
• Collaborative, supportive environment & awesome people :)

This is a hybrid role based in the San Francisco office (2 days/week)

This position is not eligible for company sponsorship.

Salary Band in U.S. (USD): $168,912-$273,368

• Comprehensive Health & Wellness: Comprehensive medical, dental, and vision coverage for you and your family, with the majority of premiums covered by Afresh. We also provide dedicated mental health support and counseling services.
• Invested in Your Future: Competitive base salary, meaningful equity (U.S. employees), and a 401(k) program with a generous company match.
• Flexible & Modern Workspace: Whether you work from home or a local office, we support your setup with a home office stipend and "Coworking Wallets" for flexible workspace access.
• Growth-Obsessed Culture: We believe in continuous learning. Every employee receives an annual professional development budget to master new skills and grow their career at Afresh.
• Holistic Monthly Stipends: Beyond your paycheck, we provide monthly stipends for "Betterment" (wellness/lifestyle) and telecommunications to ensure you have what you need to thrive.
• Time to Recharge: Flexible paid time off to take the time you need to recharge.

About Afresh

Founded in 2017, Afresh is using AI to tackle the #1 solution to curb climate change: reducing food waste. By building AI specifically for the intricacies of grocery—from the fresh perimeter to the center store—we help grocers minimize waste and maximize sales.

Afresh sits at an incredible intersection of positive social impact, rocket ship financial growth, and cutting-edge technology. Our best-in-class AI research has been published in top journals, including ICML, and our investors include Al Gore’s Just Climate, former Whole Foods Market CEO Walter Robb, and Eric Schmidt's Innovation Endeavors.

Grocery is the past, present, and future of our food system – the waste we create today will impact our planet for years to come. Join us as we continue to build a vibrant, diverse, and inclusive team that embodies our company’s values of proactivity, kindness, candor, and humility.

Afresh provides equal employment opportunities (EEO) to all employees and applicants for employment without regard to race, color, religion, sex, national origin, age, disability, genetics, sexual orientation, gender identity/expression, marital status, pregnancy or related condition, or any other basis protected by law.

Here at Afresh, many of our employees work remotely provided that they reside in one of the following states: AL, AR, CA, CO, FL, GA, IL, KY, MA, MI, MT, MO, NV, NJ, NY, NC, OR, PA, TX, WA, UT, VA, WI.