Distinguished Software Engineer
Walmart
AzureCloudDistributed SystemsGoJava
Senior Software Engineer – MLE
Walmart
full-time
Posted on:
Location Type: Office
Location: Sunnyvale • California • 🇺🇸 United States
Visit company websiteSalary
💰 $90,000 - $180,000 per year
Job Level
Senior
Tech Stack
AWSAzureCloudDockerGoogle Cloud PlatformKubernetesPythonPyTorchScikit-LearnTensorflow
About the role
- Design, develop, and deploy machine learning models at scale.
- Build and maintain end-to-end ML infrastructure for data ingestion, feature engineering, training, deployment, and monitoring.
- Develop tools and frameworks that streamline experimentation for data scientists.
- Ensure best practices in MLOps: CI/CD for ML, automated retraining, model versioning, and monitoring.
- Optimize system performance, reliability, and cost efficiency in production environments.
- Partner with data scientists to move research prototypes into robust, scalable systems.
- Collaborate with platform and infrastructure engineers to optimize compute, storage, and deployment efficiency.
- Advocate for ML engineering best practices and scalable design patterns.
- Mentor junior engineers and data scientists on production readiness and infrastructure use.
- Enable design, train, and responsible for deploying ML models or Recommendation System (ranking, retrieval, embeddings, personalization).
- Integrate recommendation outputs into a centralized agentic AI system to enhance reasoning and decision-making.
Requirements
- Option 1: Bachelor's degree in computer science, computer engineering, computer information systems, software engineering, or related area and 3 years’ experience in software engineering or related area.
- Option 2: 5 years’ experience in software engineering or related area.
- Strong expertise in MLOps and infrastructure engineering (CI/CD, monitoring, automation, Docker/Kubernetes, cloud ML platforms).
- Experience with cloud platforms (e.g., AWS, Azure, GCP).
- Proficiency in Python and ML frameworks (PyTorch, TensorFlow, Scikit-learn), plus experience with distributed data/compute systems.
- Experience in building real time and batch processed inference system.
- Strong understanding of data structures, algorithms, and statistical modeling.
- Experience with data preprocessing, feature engineering, ML training and model evaluation.
- Experience with pipeline optimization viz. Multi-processing, Multi-threading, Distributed frameworks, Caching , etc.
- Hands-on experience building and deploying ML systems at scale is preferred.
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.
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard skills
machine learningMLOpsCI/CDautomated retrainingmodel versioningPythonPyTorchTensorFlowScikit-learndata structures
Soft skills
mentoringcollaborationadvocacycommunicationproblem-solving
Certifications
Bachelor's degree in computer scienceBachelor's degree in computer engineeringBachelor's degree in computer information systemsBachelor's degree in software engineering