Member of Technical Staff, ML Research Engineer
Omnifold trains custom AI models for each customer's supply chain - purpose-built systems that forecast demand, optimize decisions, and adapt continuously to a changing world. The research team is responsible for the core intelligence that makes this possible: developing new model architectures, curating proprietary data assets, and pushing the boundaries of what ML can do.
What makes this job interesting:
You will work on problems that frontier models can't solve. Supply chain dynamics require modeling physical systems and processes.
You will own the full research cycle, from hypothesis to production model, with direct visibility into real-world impact.
You will work at the intersection of machine learning models, optimization, LLM reasoning capabilities, and proprietary data - a combination few research teams are building
What you'll own:
Training models for forecasting and optimization across complex, multi-variable supply chain environments
Building and curating proprietary data assets that carry signal about real-world physical and commercial systems
Integrating LLM knowledge and reasoning capabilities into purpose-built models to maximize accuracy and adaptability
Continuously improving model performance as market conditions shift (consumer sentiment, product launches, geopolitical changes, competitive dynamics)
What we're looking for:
5+ years of industry machine learning engineering, including and experimentation
Experience with time-series forecasting, mathematical modeling, optimization, or related domains
Understanding of LLMs, including fundamentals and practical system design including tool use and eval design
Experience working with messy, heterogeneous real-world data
Experience working with large code bases
Academic or industry research experience preferred
Comfort operating in a fast-moving, early-stage environment where research directly feeds production systems
Location: San Francisco (in-person, 5 days per week)
Omnifold’s Mission
Every bad forecast has a physical consequence. Unnecessary goods are manufactured, shipped, and stored. Emergency air freight is needed for misallocated products. Poor production planning means workers show up with nothing to do, or work frantic overtime. Inefficiency is everywhere.
Our mission is to eliminate waste and accelerate growth for every company with physical products.