AI Engineer II
We are building the next generation of hyper-personal AI systems that fundamentally transform how humans interact with businesses. Led by our Chief AI Scientist, the Personal Superintelligence Lab at Gopuff develops cutting-edge agentic AI solutions that reason about complex user contexts, preferences, and real-world constraints to deliver superhuman performance in personalized shopping assistance. You will advance the state-of-the-art in prediction, alignment, grounding, and multi-agent orchestration while deploying these breakthroughs at massive scale.
You will own core components of our instant shopping personal intelligencecontext engineering, grounding, alignment, and servingso technology accelerates customers to their goals, leveraging our unparalleled delivery speed. You'll design context engineering strategies and real-time retrieval, build SFT pipelines, and implement reinforcement learning from human feedback (RLHF) and verifiable rewards (RLVR) using GRPO / GSPO. You'll use relevant APIs for data augmentation and ship low-latency, reliable inference (batching, caching, quantization, streaming) with strong observability and safety. In close partnership with Engineering and Data Science, you'll run experiments end-to-end and deliver production models that drive measurable lifts in business KPI under privacy-by-design and rigorous evaluation.
Research Areas & Contributions
Advanced Context & Grounding Research :
- Pioneer novel architectures for multi-modal context integration across temporal, spatial, and behavioral dimensions
- Develop foundational techniques for real-time knowledge grounding with dynamic constraint satisfaction
- Research declarative programming paradigms (DSPy) for robust prompt compilation and systematic LLM behavior specification
- Design and develop multi-agent orchestration systems (LangChain, LangGraph, CrewAI, Autogen, LlamaIndex) that exhibit emergent reasoning capabilities
Alignment & Learning Systems :
Advance reasoning-centered supervised fine-tuning methodologies with novel data curation, synthetic generation, and quality assurance frameworksDevelop and evaluate model reasoning (task decomposition, tool-use, and self-correction) to improve grounded recommendations and task completion under real-world constraintsResearch parameter-efficient adaptation techniques including LoRA variants, Text-to-LoRA synthesis, and dynamic adapter routingPioneer RLHF and RLVR with emphasis on scalable oversightDevelop next-generation policy optimization algorithms beyond GRPO / GSPO with formal safety guaranteesSafety & Robustness :
Research interpretability, controllability, and alignment verification for production agentic systemsDevelop formal methods for safe exploration, reward hacking prevention, and distributional robustnessPioneer privacy-preserving techniques and federated learning approaches for personal AI systemsSystems & Deployment :
Research efficient inference architectures including novel quantization, caching, and streaming paradigmsDevelop evaluation frameworks that bridge offline metrics with online performance and safety criteriaBuild foundational infrastructure for rapid experimentation and deployment of research breakthroughsRequirements :
MSc or PhD in Computer Science, Machine Learning, or equivalent research experience with significant contributions to AI / ML literatureEstablished experience in large-scale machine learning research with demonstrated impact on real-world systemsDeep expertise in transformer architectures, large language models, and modern pre- and post-training paradigmsMastery of advanced fine-tuning techniques including LoRA / QLoRA, adapter methods, and parameter-efficient transfer learningResearch experience with agentic AI frameworks, multi-agent systems, and declarative programming approaches (DSPy, LangChain ecosystem)Strong systems engineering capabilities with PyTorch, distributed training, and cloud-native ML infrastructureTrack record of publications in top-tier venues (NeurIPS, ICML, ICLR, AAAI) or equivalent industry impactCommitment to responsible AI development and alignment researchPreferred Qualifications :
Research background in reinforcement learning, multi-agent systems, or decision theoryExperience with formal verification, program synthesis, or automated reasoningContributions to open-source AI frameworks or foundational model developmentBackground in cognitive science, human-computer interaction, or behavioral economicsExperience with privacy-enhancing technologies, federated learning, or on-device AIWhat We Offer :
Opportunity to shape the future of personal AI and human-machine collaborationAccess to world-class compute resources and datasetsCollaboration with leading researchers across academia and industryPublication support and conference travel for research disseminationCompetitive compensation with significant equity upside in breakthrough AI applicationsCompensation
Gopuff pays employees based on market pricing and pay may vary depending on your location. The salary range below reflects what we'd reasonably expect to pay candidates. A candidate's starting pay will be determined based on job-related 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 role's 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 Gopuff's equity incentive plan.Base Salary Range : $175,000 - $220,000Benefits Overview
Medical / Dental / Vision Insurance401(k) Retirement Savings PlanHSA or FSA eligibilityLong and Short-Term Disability InsuranceMental Health BenefitsFitness Reimbursement Program25% employee discount & FAM MembershipFlexible PTOGroup Life InsuranceEAP through AllOne Health (formerly Carebridge)The only predictable thing about life is that it's wildly unpredictable. That's where we come in. When life does what it does best, customers turn to Gopuff to deliver their everyday essentials, and to get through their day & night, work day and weekend. We're assembling a team of thinkers, dreamers & risk-takers...the kind of people who know the value of peace of mind in an unpredictable world. (And people who love snacks.) Like what you're hearing? Welcome to Gopuff. The Gopuff Fam is 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. We are an equal employment opportunity employer.
