Staff Data Scientist R&D

Avathon; Bengaluru; On-site; INR 3,500,000 - 5,500,000; Full time

Job summary

Avathon is seeking a Staff Data Scientist in India to design, fine-tune, and deploy small language models powering supply chain AI features. You’ll work on logistics, planning, inventory, and forecasting, building data-grounded, production-ready solutions with domain experts.

You’ll own how models appear in products, collaborating with engineering and product teams on end-to-end AI features that drive measurable enterprise value.

Qualifications

• Experience with SLMs and parameter-efficient fine-tuning for accurate NLQ with low hallucinations.
• Strong understanding of supply chain data and workflows including ERP.
• Familiarity with Knowledge Graphs and Neo4j for relationship insights.
• Python, PyTorch/TensorFlow, GraphQL, and vector databases hands-on.
• 8+ years in forecasting, optimization, or AI-driven products.
• Design of Retrieval-Augmented Generation pipelines for enterprise search.
• FAISS vector indexes tuning for low-latency deployment.
• Experience moving models from prototype to production.
• Bonus: RL/MARL, graph neural networks, or probabilistic modeling.
• Clear communication and strong product ownership.

Responsibilities

• Build and fine-tune SLMs/LLMs for production use in supply chain apps.
• Embed domain knowledge from supply chain data into models.
• Enable agentic workflows for planning, optimization, and decision support.
• Integrate models into products via Lambdas and graph-based microservices.
• Apply vector search and semantic reasoning for complex relationships.
• Work with ERP, logistics, inventory, supplier and contract data.
• Deploy and operate models with MLOps on cloud platforms.
• Measure quality and reduce hallucinations through evaluation.
• Collaborate with engineering, product, and operations to ship features.

Skills

SLMs expertise

Parameter-efficient fine-tuning

Python

PyTorch/TensorFlow

Vector databases

MLOps

Forecasting

Supply chain analytics

Graph knowledge

Knowledge Graphs

Tools

Neo4j

FAISS

GraphQL

AWS/GCP/AZURE

Job description

Who We Are & Why Join Us

Avathon is the leading Industrial AI autonomy platform, helping customers across heavy industries -- energy, mining, manufacturing, aerospace, defense, and logistics -- accelerate the journey toward autonomous operations. Our platform is built on a Computational Knowledge Graph foundation that contextualizes and connects operational data across siloed systems, bringing together time series, structured, unstructured, and machine vision data to power AI-driven applications in asset performance management, supply chain intelligence, visual AI, and global trade management. With capabilities spanning digital twins, normal behavior modeling, natural language processing, and computer vision, Avathon delivers real-time predictive intelligence and agentic decision-making at industrial scale.

Cutting-Edge AI Innovation

Join a team at the forefront of AI, developing groundbreaking solutions that shape the future.

High-Growth Environment

Thrive in a fast-scaling startup where agility, collaboration, and rapid professional growth are the norm.

Meaningful Impact

Work on AI-driven projects that drive real change across industries and improve lives.

Learn more at: avathon.com

Staff Data Scientist – Supply Chain AI (Small Language Models)

At Avathon, we build AI products that solve real industrial problems at scale. We believe the future of enterprise AI lies in small, focused, high-performance models—deeply grounded in domain context—not just large, generic systems.

We’re looking for an experienced Data Scientist to join our Supply Chain AI team in India. In this role, you’ll design, fine-tune, and deploy optimized Small Language Models (SLMs) that power production‑grade AI features across logistics, planning, inventory, and demand forecasting.

What Makes This Role Unique

This is a product-first data science role. You won’t just build models—you’ll own how they show up in real products used by large enterprises. Your work becomes the intelligence layer that sits on top of our graph and indexing systems and directly drives customer outcomes.

What You’ll Do

• Build and fine‑tune SLMs/LLMs for production use cases such as semantic search, forecasting, contract intelligence, and conversational insights. Optimize models for performance, cost, and latency using PEFT, quantization, and efficient inference techniques
• Embed supply chain knowledge into models by working closely with domain experts and product teams
• Enable agentic workflows, allowing 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 complex supply chain relationships
• Work with real enterprise data—ERP, logistics, inventory, supplier, and contract datasets
• Deploy and operate models using strong MLOps practices on platforms like AWS, GCP or AZURE
• Measure and improve quality, reducing hallucinations and improving reliability through continuous evaluation
• Partner with engineering, product, and operations teams to ship impactful AI features end-to‑end

What We’re Looking For

• Experience working with SLMs (Mistral, Phi, Nemotron, Llama variants), applying parameter‑efficient fine‑tuning (LoRA/QLoRA) to optimize task‑specific natural language query answering with high accuracy and minimized hallucinations.
• Strong understanding of supply chain data and workflows (logistics, inventory, procurement, ERP systems)
• Familiarity with Knowledge Graphs and graph‑based data modeling, including designing and querying Neo4j graphs for relationship‑driven insights.
• 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 using document chunking, metadata filtering, hybrid (dense + sparse) retrieval, and SLM/LLM grounding for enterprise knowledge search.
• Hands‑on experience building and tuning FAISS-based vector indexes (IVF, HNSW, PQ) for large‑scale semantic similarity search with low‑latency production deployment
• Experience taking models from prototype to production
• Bonus: reinforcement learning, MARL, graph neural networks, or probabilistic modeling
• Clear communication skills and a strong sense of product ownership

Why Avathon

You’ll work on core AI products, not experiments—solving complex supply chain problems for global enterprises, while building from India as a key innovation hub.

If you enjoy shipping AI that delivers measurable business value, this role is for you.

Avathon is an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, age, pregnancy, genetic information, disability, status as a protected veteran, or any other protected category under applicable federal, state, and local laws.

Avathon is committed to providing reasonable accommodations throughout the recruiting process. If you need a reasonable accommodation, please contact us to discuss how we can assist you.