Staff Engineer – Kafka, PaaS Open-Source Messaging
GEICO
AnsibleAWSAzureCloudDistributed SystemsGoGoogle Cloud PlatformGrafanaJavaKafkaPrometheusPulsar+4 more
Senior Software Engineer – MLE
Walmart
full-time
Posted on:
Location Type: Hybrid
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.
- Integrate recommendation outputs into a centralized agentic AI system to enhance reasoning and decision-making.
Requirements
- Bachelor’s/Master’s/Ph.D. in Computer Science, Machine Learning, or a related field.
- 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 systems.
- 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.
- Excellent problem-solving and analytical skills.
- Strong communication and collaboration skills.
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 preprocessing
Soft skills
problem-solvinganalytical skillscommunicationcollaboration
