Senior Machine Learning Engineer

About the role

This role is a high-ownership, production-first position within Oway, an industrial automation and research company focused on improving the efficiency of America's freight network through AI-powered solutions.

Responsibilities

• Own the development, training, and productionization of the ML systems powering Oway's pricing, routing, and matching engine, including our optimization models.
• Build and iterate on Juno AI, our autonomous LLM-based dispatch agent, including reinforcement learning workflows, prompt engineering, and edge-case handling in live operations.
• Design and maintain the data infrastructure and pipelines that feed our ML systems, from ELD telemetry ingestion to feature engineering, model monitoring, and observability.
• Collaborate closely with engineering, operations, and carrier teams to translate real-world freight problems into reliable, scalable ML solutions.
• Instrument model performance in production using LangFuse and similar tools, and iterate rapidly based on live data.
• Contribute to ML architecture decisions as an early senior hire with significant ownership.

Requirements

• 4-7 years of experience in ML engineering with a strong track record of shipping models to production.
• Proficient in Python, with hands-on experience across the ML stack, from pipeline design to model serving.
• Experience with tree-based or ensemble methods (e.g. Cubist, XGBoost) and/or LLM fine-tuning and RLHF workflows.
• Strong understanding of data pipeline design, including ETL, feature stores, streaming ingestion, or similar.

Qualifications

• Energized by unstructured problems in physical systems, and motivated by real-world impact over research novelty.
• Bonus: experience in logistics, routing, geospatial ML, or agent-based systems.
• Bonus: familiarity with LangChain, LangFuse, or similar LLM observability tooling.

Skills

• JavaScript, Python, TypeScript, SQL, Torch/PyTorch, Natural Language Processing, ETL

Benefits

Oway offers competitive compensation, including a salary range of $160K - $250K, benefits, and a remote work option in San Francisco, CA, US.

Pay

$160K - $250K

Schedule

Full-time