Senior AI Engineer — omniXM (AI Platform)
Location: Pune - On-Site
Type: Full time
Team: AI Engineering
Reporting To: Head of Engineering
ABOUT OMNIXM
omniXM builds intelligent customer experience management products. Our flagship AI product omniXMOS, includes a conversational survey analysis engine omniSense, that processes real-world business feedback at scale. The system combines deterministic data tools, LLM-powered reasoning, multi-modal report generation, and document-grounded RAG — all served through a FastAPI backend and deployed on Azure.
We are growing our AI Engineering team to deepen the intelligence layer of omniSense: smarter agents, richer retrieval, better-adapted models, and tighter orchestration.
ROLES & RESPONSIBILITIES
1. Agentic AI Design & Orchestration
· Design and implement multi-agent pipelines for survey analysis, insight generation, and automated reporting using frameworks such as LangGraph, CrewAI, or custom orchestration built on FastAPI.
· Build specialized sub-agents (e.g., a data-fetch agent, a metrics-aggregation agent,
a narrative-synthesis agent, a report-formatting agent) that coordinate under a planner/orchestrator pattern.
· Implement tool calling that bridges LLM reasoning with deterministic backend tools — including ticket summaries, dimension breakdowns, order/menu extraction, and period comparisons.
· Add Human-in-the-Loop (HITL) checkpoints and agent memory (session context, cross-turn state) to support long-running, multi-step analysis workflows.
· Maintain fault-tolerant execution: retry logic, graceful degradation, and async background task handling for slow or multi-step agentic flows.
2. Retrieval-Augmented Generation (RAG)
· Own and extend the document RAG pipelines, including chunking strategies, embedding model selection, vector store management (currently SQLite-backed, with a path to Milvus/Qdrant at scale), and retrieval tuning.
· Design hybrid retrieval combining dense vector search with keyword/BM25 re-rankin
📌 Senior AI Engineer (Pune)
🏢 omniXM
📍 Pune