[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 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