Staff Data Scientist R&D

The role

You will design, fine-tune, and deploy optimized Small Language Models to power AI features for logistics, planning, inventory, and demand forecasting. You will work on the Supply Chain AI team to integrate models into products using graph-based microservices architecture.

What you will do

• Build and fine-tune SLMs/LLMs for production use cases
• Optimize models for performance, cost, and latency using PEFT, quantization, and efficient inference techniques
• Embed supply chain knowledge into models by working with domain experts and product teams
• Enable agentic workflows for AI Assistants to execute planning, optimization, and decision-support tasks
• Integrate models into products using custom Lambdas and a Graph-based microservices architecture
• Apply vector search and semantic reasoning to help customers navigate supply chain relationships
• Work with enterprise data including ERP, logistics, inventory, supplier, and contract datasets
• Deploy and operate models using MLOps practices on AWS, GCP or AZURE
• Measure and improve quality by reducing hallucinations and improving reliability
• Partner with engineering, product, and operations teams to ship AI features

Required qualifications

• Experience working with SLMs (Mistral, Phi, Nemotron, Llama variants)
• Applying parameter-efficient fine-tuning (LoRA/QLoRA) to optimize task-specific natural language query answering
• Strong understanding of supply chain data and workflows (logistics, inventory, procurement, ERP systems)
• Familiarity with Knowledge Graphs and graph-based data modeling
• Designing and querying Neo4j graphs
• Solid hands-on skills in Python, PyTorch/TensorFlow, GraphQL, and vector databases
• 8+ years of experience in forecasting, optimization, supply chain analytics, or AI-driven products
• Experience designing Retrieval-Augmented Generation pipelines
• Hands-on experience building and tuning FAISS-based vector indexes
• Experience taking models from prototype to production
• Clear communication skills
• Strong sense of product ownership

Nice to have

• reinforcement learning
• MARL
• graph neural networks
• probabilistic modeling

About the company

Avathon is an Industrial AI autonomy platform helping customers in heavy industries accelerate the journey toward autonomous operations. The platform uses a Computational Knowledge Graph foundation to power AI-driven applications.