AI Engineer III

Gopuff is seeking an AI Engineer III to lead the design and deployment of agentic AI systems focused on real-world context and constraints, leveraging up‐to‐the‐minute knowledge and unparalleled delivery speed. The role involves defining architecture, aligning research with real-world applications, mentoring colleagues, and scaling systems in production. It's a leadership position focused on safety, privacy, and robust systems engineering.

Requirements

• PhD in Computer Science, Machine Learning, or equivalent research experience
• 7+ years of experience building and shipping large-scale ML systems with significant ownership
• Mastery of advanced fine-tuning techniques (LoRA, QLoRA, Adapter methods, parameter-efficient transfer learning)
• Research experience with agentic AI frameworks, multi-agent systems, and declarative programming (DSPy, LangChain ecosystem)
• Strong systems engineering capabilities with PyTorch, distributed training, and cloud-native ML infrastructure
• Track record of publications in top-tier venues
• Deep expertise in transformer architectures, SFT, and RLHF
• Strong grounding in offline evaluation, counterfactual estimators, and safe online ramp strategies
• Systems fluency: PyTorch, distributed training, low‐latency serving, observability, and cloud‐native ML infra

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