Senior Data Scientist

Job Summary

We are looking for a Senior Data Scientist to lead the development of scalable Generative AI pipelines that process raw data and generate context-aware results to power Plum’s AI-driven products. You will play a central role in shaping our GenAI platform, working across the full ML lifecycle—from ingestion and retrieval to generation, evaluation, and deployment.

This role combines deep expertise in machine learning with hands-on experience in building production-grade systems. You’ll collaborate closely with various cross functional teams and operate in a fast-paced environment where innovation, autonomy, and ownership are key.

Key Responsibilities

• Design and architect end-to-end Generative AI pipelines using LLMs to process and generate context-aware results.
• Integrate open-source and proprietary LLMs (e.g., GPT, LLaMA) via APIs and custom orchestration.
• Build and optimize workflows using frameworks such as LangChain
• Design and implement RAG (Retrieval-Augmented Generation) architecture to inject relevant, contextual data into generation prompts.
• Develop robust methods to evaluate and compare LLM outputs based on relevance, personalization, and factual accuracy.
• Build automated and scalable LLM evaluation pipelines using embedding-based similarity, scoring metrics, and human-in-the-loop feedback.
• Implement monitoring, observability, and logging for GenAI workflows to ensure reliability in production.
• Collaborate with cross-functional teams to integrate generative outputs into client-facing applications.