[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 roadmapSkills• 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 technologyBenefits• 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 isCompany 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