[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 Apply tot his job Apply To this Job