Distinguished, Data Scientist - Agentic AI Systems Engineering & Model Post-Training

Distinguished Data Scientist - Agentic AI Systems Engineering & Model Post-Training

This role exists to build new generation production-grade agentic AI systems that reason over complex enterprise information, coordinate specialized agents, use tools safely, plan and execute long-horizon work, and continuously improve through rigorous evaluation and model post-training. This is a deeply hands-on AI systems engineering position focused on designing, building, and operating production software. As the Distinguished technical leader in this space, you will own production systems end to end across the full technology stack—agent orchestration and model logic, backend services and APIs, the data layer, and web and CLI/TUI interfaces.

You will build advanced agentic systems end to end, own the architecture, implementation, evaluation, launch, and operation of complex, autonomous multi-agent applications that tackle Walmart's highest-value enterprise initiatives. You will build systems in which specialized agents can reason, plan across long horizons, coordinate work, use tools, retrieve enterprise knowledge, maintain secure and durable session and workflow state, manage context growth, ask for human approval when needed, and cancel, resume, replay, or recover safely from failure.

You will own the design and build of advanced multi-agent harnesses, runtimes, and orchestration capabilities using frameworks such as Pydantic AI, Lang Graph, Lang Chain, Auto Gen, or Llama Index, or purpose-built custom infrastructure. You will improve model reasoning and quality through hands-on post-training: reinforcement learning (RLHF/RLAIF), preference optimization, supervised fine-tuning, distillation, and reward and grader design to improve reasoning, tool-use reliability, and domain-specialized behavior across frontier models and smaller, domain-specialized models.

You will make every solution safe, scalable, and production-ready by designing systems on Google Cloud Platform (GCP) and Walmart internal technologies that are secure, fault tolerant, cost-aware, observable, highly available, and low latency. You will lead by building and raising the bar through clear technical writing, rigorous code and architecture reviews, reusable platform capabilities, and hands-on mentoring of Principal, Staff, and senior engineers.

The bar for this role requires 10+ years of hands-on software engineering and architecture experience; 12+ years preferred. You must have demonstrated Distinguished, Fellow, Principal, Senior Staff, or equivalent scope: you have led difficult technical work across organizations and taken direct, hands-on ownership of shipping and operating complex systems. You must have co-equal depth in agentic AI engineering, software engineering, and machine learning engineering: ML and LLM fundamentals, experimental design, retrieval and ranking, model evaluation, production trade-offs, and sound judgment about when an LLM is not the right solution.