Build and deploy production ML systems end-to-end: data pipelines, feature stores, model training, serving layers, monitoring, and feedback loops at scale.
Provide technical vision and lead advanced development in agentic AI, reinforcement learning, and simulation.
Architect and deploy large-scale AI systems and autonomous AI agents — conversational assistants, predictive agents with tool-calling, and multi-agent orchestration systems using LangChain, Pydantic AI, and RAG patterns.
Deploy NLP pipelines at enterprise scale — semantic similarity across millions of records using BERT/SBERT embeddings and vector search (FAISS).
Lead high-impact projects end-to-end; problem framing, modeling, prototyping, production architecture, and rollout.
Mentor senior scientists and engineers on advanced modeling techniques, feature engineering, and experimentation design
raising the technical bar across teams.
Requirements
Experience building and deploying AI/ML solutions in production environments.
Strong profiency in Python and common ML libraries (scikit-learn, XGBoost/LightGBM, PyTorch/TensorFlow).
Demonstrated AI/ML engineering experience: Pipeline orchestration (Airflow, Dagster, Prefect) Model packaging/serving (FastAPI, BentoML, TorchServe) Containerization and deployment (Docker, Kubernetes) CI/CD and testing practices for AI/ML Monitoring, observability (model metrics, drift, loggin)
Solid understanding of statistics, experimental design, and causal/measurement thinking.
Experience with agentic AI / LLM systems including: Tool/function calling and agent orchestration framework (LangGraph/LangChain, Semantic Kernel, LlamaIndex, custom orchestrators).
Strong SQL skills and experience working with large-scale datasets.
Ability to translate complex analytical results into clear business recommendations for non-technical stakeholders.
Minimum Qualifications
Masters degree in Statistics, Economics, Analytics, Mathematics, Computer Science, Information Technology, or related field AND 3+ years’ experience in data science or Machine Learning roles with demonstrated impact.
Tech Stack
Airflow
Docker
Kubernetes
Python
PyTorch
Scikit-Learn
SQL
Tensorflow
Benefits
Health benefits include medical, vision and dental coverage.
Financial benefits include 401(k), stock purchase and company-paid life insurance.
Paid time off benefits include PTO (including sick leave), parental leave, family care leave, bereavement, jury duty, and voting.
Other benefits include short-term and long-term disability, company discounts, Military Leave Pay, adoption and surrogacy expense reimbursement, and more.
Live Better U is a Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities. Programs range from high school completion to bachelor's degrees, including English Language Learning and short-form certificates. Tuition, books, and fees are completely paid for by Walmart.