[Remote] Principal Machine Learning Engineer - Consumer Personalization & Generative AI

Note: The job is a remote job and is open to candidates in USA. Gopuff’s Data Science team is expanding its mission to include cutting-edge generative-AI capabilities. This role will lead the design and deployment of LLM-driven models to enhance customer experiences through personalized recommendations and timely interactions.

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

Skills

• 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

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)

Company Overview

• Gopuff is a digital delivery service designed to deliver daily essentials within minutes. It was founded in 2013, and is headquartered in Philadelphia, Pennsylvania, USA, with a workforce of 5001-10000 employees. Its website is http://www.gopuff.com.

Company H1B Sponsorship

• Gopuff has a track record of offering H1B sponsorships, with 9 in 2025, 12 in 2024, 27 in 2023, 52 in 2022, 18 in 2021, 5 in 2020. Please note that this does not guarantee sponsorship for this specific role.

Apply tot his job Apply To this Job