<div>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.</div>The ML Platform Engineering team at Afresh is responsible for building and maintaining the foundational infrastructure and tooling that powers all of our machine learning and applied science solutions. We provide the shared components and services that enable our teams to develop, deploy, and scale robust ML models. This includes a performant data API, configurable featurization, reliable forecasting systems, highly parallel optimization engines, and scalable training pipelines, and deep experimentation capabilities. As our product suite and customer base grow, so does the scale and complexity of what our platform needs to support, gracefully accommodating predictions and simulations across various time scales (hours, days, weeks), complex data hierarchies (pallets on a truck, shelves of mangos in a store, chunks of fruit in a bowl), and endless configuration possibilities (average shelf fullness, backroom loads, truck capacities).About the RoleAs a Staff ML Platform Engineer on the ML Platform Engineering team, you will be instrumental in elevating our core ML platform to its next level of performance, reliability, and scalability. You'll work on the critical infrastructure that directly enables all of Afresh's Machine Learning and Applied Science teams to innovate faster and deliver impact. Your contributions will empower our product suite, including our flagship Prediction Engine, to power replenishment decisions on more than 15% of all produce sold in the United States.What You'll Do:<ul><li>In your first 3 months, you'll partner with ML, Applied Science, and engineering leadership to identify the highest-leverage gaps in our ML platform and shape a multi-quarter technical strategy to close them.</li><li>By the end of your first 6 months, you will have driven a platform-level initiative that meaningfully changes what's possible at Afresh — establishing the architecture for real-time inference across the company, redesigning model configuration and deployment end-to-end, or rebuilding our distributed inference layer for an order-of-magnitude growth in scale — while raising the bar across the ML org through design reviews and mentorship of other senior engineers.</li></ul>Skills and Experience<ul><li>BS in Computer Science or a relevant technical field.</li><li>7+ years of professional software development experience with a proven track record of shipping high-quality applications and services.</li><li>Experience working collaboratively with machine learning engineers, data scientists, or applied scientists on large-scale software projects involving machine learning models.</li><li>You possess a genuine curiosity about ML modeling (e.g., demand forecasting, state estimation, ordering policy). You aren't just building "pipes"; you want to understand what is flowing through them. </li><li>You have a deep understanding of how scientists work and build tools that bridge the gap between a research notebook and production-grade software. </li><li>Technical leadership experience and a demonstrated ability to mentor junior engineers.</li><li>Deep expertise in library design, API design, data structures, and algorithms.</li><li>Strong familiarity with the Python ecosystem (NumPy, Pandas, Torch, PySpark). While our stack is Python-heavy, we value engineers who are stack-agnostic and focus on solving fundamental distributed systems problems. </li><li>Proven ability to architect high-throughput distributed inference systems using Spark, Dask, or Ray. </li><li>Experience in engineering robust data architectures with a focus on schema evolution and performance tuning. </li><li>Experience with end-to-end orchestration (lineage tracking, resource scheduling, self-healing pipelines).</li></ul>Tech Stack: Our backend is pure Python (NumPy, Pandas, Torch, PySpark, Cython, orchestrated in Airflow). We use Databricks as our data warehouse. While we'd like you to have very good familiarity with Python, many of our problems are stack-agnostic.This position is not eligible for company sponsorship.Salary Band in U.S. (USD): $169,000 - $273,000 <div>About AfreshFounded 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.</div>

Afresh is on a mission to reduce food waste and increase access to nutritious food globally by transforming the fresh food supply chain. Our AI-powered solutions optimize the multi-trillion-dollar grocery industry's food ordering, production, and merchandising processes. We built the first platform capable of handling all of Fresh food&rsquo;s complexities.
 We are relentlessly focused on helping grocers reduce waste, increase profitability, and bring better, fresher food to their communities. Afresh is currently deployed in hundreds of grocery stores across the United States and is on track to reduce 30 million pounds of food waste per year with our existing customers alone. By 2021, we're aiming to reduce food waste by a quarter billion pounds per year.
We are building a vibrant, diverse, and inclusive team that embodies our company&rsquo;s values: proactivity, kindness, candor, and humility. We aspire to continually grow as individuals and as an organization to live these values and realize our mission. We earnestly believe that Afresh represents a one-of-a-kind opportunity to have massive social impact at scale by employing novel technology&mdash;and to have a ton of fun along the way.

Staff Software Engineer, ML Platform

About the Job

About the Company

Afresh Technologies

Similar Jobs

Senior Machine Learning Infrastructure Engineer

Senior Machine Learning Infrastructure Engineer

Software Engineer, ML Infrastructure

Software Engineer, ML Infrastructure

Software Engineer, ML Platform

Software Engineer, ML Platform

Staff Machine Learning Engineer

Staff Machine Learning Engineer