Principal Data Scientist - Consumer Personalization & Generative AI

Gopuff • Boston, MA

70d • $175,000 - $275,000

About The Position

Gopuff’s Data Science team powers the personalized search, browse and recommendation experiences that shape how customers discover daily goods. We are expanding our mission to include cutting-edge generative-AI capabilities. This role will lead the design and deployment of LLM-driven models that deeply understand customer needs and deliver personalized, timely experiences. You will partner with product managers, engineers and researchers to adapt pre-trained large language models (LLMs) to Gopuff’s unique business challenges, fine-tune models using LoRA/qLoRA, build retrieval-augmented generation (RAG) pipelines, and drive innovation from prototype through production.

Requirements

• MS/PhD in Computer Science, Statistics, Mathematics or a related field with 2+ years of experience building generative-AI systems (or 8+ years of industry experience in data science).
• Demonstrated experience fine-tuning LLMs using LoRA/qLoRA, adapting models through RAG pipelines, and optimizing with transformer architectures.
• Proficiency in prompt engineering, RLHF and model optimization.
• Hands-on expertise with Hugging Face Transformers, LangChain, LlamaIndex and vector databases (e.g., Pinecone, FAISS).
• Strong skills in Python and ML frameworks (PyTorch, TensorFlow, JAX); experience with cloud platforms (AWS, GCP, Azure) and MLOps tools (MLflow, Databricks, etc.).
• Experience working with large datasets and SQL; ability to write scalable, production-quality code.
• Excellent communication and collaboration skills to partner across technical and business teams.
• A passion for building AI that understands and anticipates customer needs while ensuring responsible and fair use of technology.

Responsibilities

• Understand customer needs through AI: Build models that interpret intent and context to provide the right product, recommendation or experience at the right moment.
• Fine-tune and adapt LLMs: Apply LoRA/qLoRA and RAG techniques to customize state-of-the-art LLMs for Gopuff’s domain, using frameworks such as Hugging Face, LangChain and LlamaIndex.
• Architect personalization systems: Design robust pipelines for search, retrieval and ranking that combine generative AI with classical ML, leveraging vector databases and prompt/context optimization.
• Drive algorithmic innovation: Explore and apply new methods in deep learning, reinforcement learning, multi-task learning and embeddings to improve personalization, recommendations and discovery.
• Develop conversational and Q&A experiences: Build interactive agents that can answer questions, guide product discovery and engage with customers in natural language.
• Deploy at scale: Collaborate with MLOps and engineering teams to productionize models with high scalability, reliability and low-latency response times.
• Evaluate and optimize: Define metrics, design A/B tests and perform offline/online evaluations to measure model performance, customer satisfaction and business impact.
• Collaborate and educate: Work with product, engineering, design, analytics and leadership teams to translate business needs into AI solutions and communicate data-science concepts clearly.
• Mentor and lead: Provide technical leadership and mentorship to data scientists and ML engineers, fostering a culture of responsible, customer-focused AI.
• Stay ahead of the curve: Continuously track emerging LLM architectures, tools and techniques, and integrate them into Gopuff’s AI roadmap.

Benefits

• Medical/Dental/Vision Insurance
• 401(k) Retirement Savings Plan
• HSA or FSA eligibility
• Long and Short-Term Disability Insurance
• Mental Health Benefits
• Fitness Reimbursement Program
• 25% employee discount & FAM Membership
• Flexible PTO
• Group Life Insurance
• EAP through AllOne Health (formerly Carebridge)

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: Master's degree