W

Walmart

Sells groceries and essentials in stores and online

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

AI Application Engineering +1Confirmed live in the last 24 hours
Full-timeBentonville, AR, USAIn-person

About the Job

Requirements

  • 10+ years of hands-on software engineering and architecture experience; 12+ years preferred. Demonstrated Distinguished, Fellow, Principal, Senior Staff, or equivalent scope, including leading difficult technical work across organizations and hands-on ownership of shipping and operating complex systems.
  • Exceptional end-to-end builder ownership across agent runtime, CLI/TUI, web frontend, backend services and APIs, and the underlying data layer; ability to take an ambiguous operational problem from first principles through prototype, trade-offs, stakeholder trust, and production hardening.
  • Co-equal depth in agentic AI engineering, software engineering, and machine learning engineering, including machine learning and large language model fundamentals, experimental design, retrieval and ranking, model evaluation, production trade-offs, and judgment about when a large language model is not the right solution.
  • Deep hands-on model post-training experience, including reinforcement learning from human or AI feedback, preference optimization, supervised fine-tuning, distillation, and reward and grader design to improve reasoning, tool-use reliability, and domain-specialized behavior, including smaller domain-specialized models.
  • Deep production experience with stateful, policy-controlled agent runtimes and tool-using multi-agent systems for autonomous long-horizon work, including durable session lifecycle, context-window management, retrieval-augmented generation, guardrails, evaluation systems, secure delegation, safe tool use, interruption handling, recovery, Model Context Protocol, and production tool and integration lifecycle management.
  • Broad applied machine learning depth across areas such as reinforcement learning, multimodal modeling, retrieval and ranking, forecasting, optimization and operations research, experimentation, or model evaluation; production machine learning lifecycle and MLOps practices; judgment about when a classical model is preferable to a large language model.
  • Expert Python and strong production backend capability in one or more of Go, Java, or TypeScript, with full-stack capability across React/TypeScript frontend, APIs, and the data layer.
  • Demonstrated depth in distributed systems and production reliability, including streaming and batch platforms, Kubernetes/Docker, continuous integration and delivery, telemetry and observability, service-level objectives and indicators, availability, latency and cost management, secure enterprise integrations, incident learning, and layered testing.
  • Deep fluency with AI-native software-development workflows and coding agents such as Claude Code, Codex, Cursor, Copilot, or equivalent tools, applying rigorous engineering judgment, code review, evaluation, and testing.
  • Evidence of exceptional hands-on depth and organization-level impact, such as shipped systems, architecture and code walkthroughs, permitted open-source contributions, technical demonstrations, publications, or patents; ability to describe measurable and sustained impact beyond one's own team. A public portfolio is welcome but not required.
  • High agency, technical creativity, intellectual rigor, sustained execution, proactive work in ambiguity, and ethical collaboration.
  • Meet one corporate eligibility path: a master's degree in Computer Science, Computer Engineering, Computer Information Systems, Software Engineering, Machine Learning, Artificial Intelligence, or a related area plus four years of experience in software engineering or a related area; or eight years of experience in software engineering or a related area. The role-specific hiring bar requires 10+ years of hands-on engineering and architecture experience.
  • Bachelor's degree in Statistics, Economics, Analytics, Mathematics, Computer Science, Information Technology, or a related field and six years of experience in an analytics-related field; or master's degree in those fields and four years of experience in an analytics-related field; or eight years of experience in an analytics or related field.

Responsibilities

  • Own the architecture, implementation, evaluation, launch, and operation of complex autonomous multi-agent applications, from full-stack proof of concept through dependable production-scale service.
  • Build systems that enable specialized agents to reason, plan over long horizons, coordinate work, use tools, retrieve enterprise knowledge, maintain secure durable session and workflow state, manage context growth, request human approval, and safely cancel, resume, replay, or recover from failure.
  • Build a policy-first agent runtime and control plane with deterministic allow, deny, and ask decisions; least-privilege tool and data access; scoped identity and authorization; auditable human approvals; bounded subagent delegation; and safe cancellation, user steering, and retry behavior.
  • Own the full-stack product path across agent and model logic, backend services and APIs, database and schema design, data modeling, migrations, data integrations, telemetry, monitoring, and React/TypeScript experiences; design AI-native CLI/TUI or headless structured-output interfaces when appropriate.
  • Build and own the Discovery Loop within the Autonomous Supply Chain Engine, including its execution model, knowledge and context layer, evaluation loop, and learning path that turns human feedback and business outcomes into measurable improvement.
  • Partner with supply chain experts, scientists, data owners, operations, and engineering stakeholders to discover workflows, prototype with real users, measure value and risk, and harden successful solutions.
  • Design and build multi-agent harnesses, runtimes, and orchestration capabilities using established frameworks or custom infrastructure; establish typed contracts, testability, scalability, operability, and developer ergonomics.
  • Design explicit, inspectable agent execution models with explicit state, typed dependencies and handoffs, branching, subagent fan-out and fan-in, retry and repair paths, loop detection, cost and time budgets, hard termination conditions, and external verification.
  • Build governed knowledge and context infrastructure, including knowledge and context graphs, vector retrieval, relational and temporal models or hybrids; own schemas and ontology, entity resolution, conflict handling, data construction and enrichment pipelines, provenance and temporal validity, schema evolution, and multi-hop retrieval.
  • Build and operate agent skills, tool adapters, hooks and extension points, Model Context Protocol clients and servers, structured outputs, function calling, enterprise APIs, identity, and secrets management; own secure configuration and authentication, tool binding, schema compatibility, discovery, timeouts, health monitoring, retries, circuit breaking, quarantine, cleanup, failure isolation, and auditable operations.
  • Create governed extension ecosystems for agents, skills, commands, plugins, hooks, tool adapters, and workflows; define stable contracts, compatibility and versioning, secure installation and updates, isolation, rollback, and observability.
  • Treat agent quality as an engineering discipline through golden tasks, offline benchmarks, online experiments, adversarial and regression testing, failure analysis, measurable quality thresholds, and release gates.
  • Establish AgentOps and LLMOps practices for prompt and tool versioning, tracing, evaluation datasets, workflow reliability, cost and latency controls, incident learning, and continuous improvement; instrument systems to trace agent actions, permissions, model/tool/policy versions, data and integrations, cost, failures, reproduction, and remediation.
  • Improve model reasoning and quality through model post-training techniques, including reinforcement learning from human or AI feedback, preference optimization, supervised fine-tuning, distillation, and reward and grader design.
  • Build the model-improvement flywheel by turning production traces, tool-use trajectories, human feedback, reasoning paths, enterprise and operational data, synthetic data, and curated evaluation sets into governed training and evaluation datasets; post-train, distill, evaluate, and redeploy specialized models while owning governance, provenance, privacy, and access controls.
  • Design provider-aware, model-agnostic execution and routing layers that account for model capabilities, multimodal inputs and outputs, context limits, structured outputs, streaming, credentials, rate limits, transient failures, and quality, latency, and cost trade-offs.
  • Design systems on Google Cloud Platform and Walmart internal technologies to be secure, fault tolerant, cost-aware, observable, highly available, and low latency.
  • Build responsible enterprise-agent behavior through least-privilege access, authorization boundaries, audit trails, data-protection controls, prompt-injection and untrusted-instruction defenses, approval controls for consequential actions, and reversible recovery paths.
  • Design distributed data, streaming, and telemetry systems using technologies such as Kafka, Flink, Spark, OpenTelemetry, and Grafana, and use operational signals to improve quality, reliability, and user trust.
  • Set quality, performance, and release standards through layered unit, integration, API, workflow, and user-journey testing; adversarial safety testing; benchmark- and profile-driven performance work; continuous integration and delivery quality gates; reliable rollout and rollback; and incident-response readiness.
  • Deliver accessible associate-facing experiences meeting applicable Walmart accessibility standards, including WCAG 2.2 AA where applicable.
  • Set multi-year technical direction for shared agent foundations and products while remaining hands-on with architecture, critical-path code, technical spikes, design reviews, and production incidents.
  • Raise the engineering bar through technical writing, code and architecture reviews, reusable platform capabilities, and hands-on mentoring of Principal, Staff, and senior engineers.
  • Make consequential technical trade-offs based on evidence, security, maintainability, measurable outcomes, and user needs.
  • Turn ambitious ideas into responsibly operated products through high standards, constructive collaboration, direct communication, and follow-through.

Desired Qualifications

  • 12+ years of hands-on software engineering and architecture experience.
  • Ph.D. in Computer Science, Computer Engineering, Machine Learning, Artificial Intelligence, or a closely related discipline.
  • Experience building developer-facing agentic platforms, software development kits, or coding and automation agents used by other engineers.
  • Graph and knowledge-representation depth, including knowledge and context graphs, ontology and schema design, entity resolution, multi-hop retrieval, graph databases and query languages, or graph machine learning and network analysis.
  • Deep experience designing and operating machine learning/large language model platforms, agent runtimes, model-serving systems, retrieval and ranking platforms, or high-scale distributed data systems.
  • Meaningful open-source leadership, technical talks, developer tools, technical writing, patents, or high-impact research publications in agentic artificial intelligence, machine learning systems, or software platforms.
  • Evidence of building innovative technologies or applications that improve complex operational workflows for frontline users, associates, customers, or other high-impact users.
  • Data science, machine learning, and optimization-model experience; Ph.D. in Machine Learning, Computer Science, Information Technology, Operations Research, Statistics, or Applied Mathematics; publications or active peer review in related journals or conferences; successful completion of assessments in Python, Spark, Scala, or R; and experience using open-source frameworks such as scikit-learn, TensorFlow, or PyTorch.
  • Background creating inclusive digital experiences, knowledge of implementing Web Content Accessibility Guidelines 2.2 AA standards and assistive technologies, and applying digital accessibility best practices.

About the Company

Walmart

Sells groceries and essentials in stores and online

Walmart sells groceries, household essentials, clothing, and electronics through stores and online marketplaces for consumers around the world. Its everyday low price model combines large purchasing volumes with distribution networks that move goods to stores and customer homes. Shoppers can browse online, collect orders at a nearby store, or request delivery, while third-party merchants use its marketplace and fulfillment services to reach customers. The group earns revenue from merchandise sales, memberships, advertising, and services, and also operates pharmacies, optical centers, and digital payments businesses. Walmart opened its first discount store in 1962 and is headquartered in Bentonville, Arkansas.

Company size
10,001+
Company stage
IPO
Headquarters
Bentonville, AR
Founded
1962

Recently Posted Jobs

Explore more opportunities at Walmart

See more jobs