Principal Machine Learning Engineer - Consumer Personalization & Generative AI (San Francisco)

Qualifications

• MS/PhD in Computer Science, Statistics, Mathematics or a related field with 2+ years of experience building generativeAI systems (or 8+ years of industry experience in data science)
• Demonstrated experience finetuning LLMs using LoRA/qLoRA, adapting models through RAG pipelines, and optimizing with transformer architectures
• Proficiency in prompt engineering, RLHF and model optimization
• Handson 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, productionquality 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

• Gopuff pays employees based on market pricing and pay may vary depending on your location
• For additional information on this roles compensation package, please reach out to the designated recruiter for this role
• This role is eligible for a discretionary annual cash bonus and participation in Gopuffs equity incentive plan
• Base Salary Range: $175,000 - $270,000
• Medical/Dental/Vision Insurance
• 401(k) Retirement Savings Plan
• HSA or FSA eligibility
• Long and ShortTerm Disability Insurance
• Mental Health Benefits
• Fitness Reimbursement Program
• 25% employee discount & FAM Membership
• Flexible PTO
• Group Life Insurance
• EAP through AllOne Health (formerly Carebridge)

Responsibilities

• This role will lead the design and deployment of LLMdriven models that deeply understand customer needs and deliver personalized, timely experiences
• You will partner with product managers, engineers and researchers to adapt pretrained large language models (LLMs) to Gopuffs unique business challenges, finetune models using LoRA/qLoRA, build retrievalaugmented generation (RAG) pipelines, and drive innovation from prototype through production
• Understand customer needs through AI: Build models that interpret intent and context to provide the right product, recommendation or experience at the right moment
• Finetune and adapt LLMs: Apply LoRA/qLoRA and RAG techniques to customize stateoftheart LLMs for Gopuffs 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 classic ML, leveraging vector databases and prompt/context optimization
• Drive algorithmic innovation: Explore and apply new methods in deep learning, reinforcement learning, multitask 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 lowlatency 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 datascience concepts clearly
• Mentor and lead: Provide technical leadership and mentorship to data scientists and ML engineers, fostering a culture of responsible, customerfocused AI
• Stay ahead of the curve: Continuously track emerging LLM architectures, tools and techniques, and integrate them into Gopuffs AI roadmap

Full Description

Gopuffs 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 cuttingedge generativeAI capabilities.

This role will lead the design and deployment of LLMdriven models that deeply understand customer needs and deliver personalized, timely experiences .

You will partner with product managers, engineers and researchers to adapt pretrained large language models (LLMs) to Gopuffs unique business challenges, finetune models using LoRA/qLoRA, build retrievalaugmented generation (RAG) pipelines, and drive innovation from prototype through production.

You Will:

• Understand customer needs through AI: Build models that interpret intent and context to provide the right product, recommendation or experience at the right moment.

• Finetune and adapt LLMs: Apply LoRA/qLoRA and RAG techniques to customize stateoftheart LLMs for Gopuffs 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 classic ML, leveraging vector databases and prompt/context optimization.

• Drive algorithmic innovation: Explore and apply new methods in deep learning, reinforcement learning, multitask 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 lowlatency 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 datascience concepts clearly.

• Mentor and lead: Provide technical leadership and mentorship to data scientists and ML engineers, fostering a culture of responsible, customerfocused AI.

• Stay ahead of the curve: Continuously track emerging LLM architectures, tools and techniques, and integrate them into Gopuffs AI roadmap.

You Have:

• MS/PhD in Computer Science, Statistics, Mathematics or a related field with 2+ years of experience building generativeAI systems (or 8+ years of industry experience in data science).

• Demonstrated experience finetuning LLMs using LoRA/qLoRA, adapting models through RAG pipelines, and optimizing with transformer architectures.

• Proficiency in prompt engineering, RLHF and model optimization.

• Handson 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, productionquality 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.

Compensation:

• Gopuff pays employees based on market pricing and pay may vary depending on your location.

The salary range below reflects what wed reasonably expect to pay candidates.

A candidates starting pay will be determined based on jobrelated skills, experience, qualifications, interview performance, and market conditions.

These ranges may be modified in the future.

Exceptions may be made for exceptional individuals. For additional information on this roles compensation package, please reach out to the designated recruiter for this role.

• This role is eligible for a discretionary annual cash bonus and participation in Gopuffs equity incentive plan.

• Base Salary Range: $175,000 - $270,000

What We Offer:

• Medical/Dental/Vision Insurance

• 401(k) Retirement Savings Plan

• HSA or FSA eligibility

• Long and ShortTerm Disability Insurance

• Mental Health Benefits

• Fitness Reimbursement Program

• 25% employee discount & FAM Membership

• Flexible PTO

• Group Life Insurance

• EAP through AllOne Health (formerly Carebridge)

At Gopuff, we know that life can be unpredictable.

Sometimes you forget the milk at the store, run out of pet food for Fido, or just really need ice cream at 11 pm.

We get itstuff happens.

But thats where we come in, delivering all your wants and needs in just minutes.

And now, were assembling a team of motivated people to help us drive forward that vision to bring a new age of convenience and predictability to an unpredictable world.

Like what youre hearing?

Then join us on Team Blue.

#LI-GOPUFF

Gopuff is an equal employment opportunity employer, committed to an inclusive workplace where we do not discriminate on the basis of race, sex, gender, national origin, religion, sexual orientation, gender identity, marital or familial status, age, ancestry, disability, genetic information, or any other characteristic protected by applicable laws. We believe in diversity and encourage any qualified individual to apply.

#J-18808-Ljbffr