Walmart
Company
(IND) Staff, Machine Learning Engineer
Job Description
Position Summary...
What you'll do...
Walmart’s Enterprise Business Services (EBS) is a powerhouse of exceptional teams delivering world-class technology solutions and services that make a profound impact at every level of Walmart.
As a key part of Walmart Global Tech, our teams set the bar for operational excellence and leverage emerging technologies to support millions of customers, associates, and stakeholders worldwide. Each time an associate turns on their laptop, a customer makes a purchase, a new supplier is onboarded, the company closes its books, and associates are paid accurately and on time — that’s EBS.
Joining EBS means embarking on a journey of limitless growth, relentless innovation, and the opportunity to shape the future of Walmart.
The EBS Data & AI team is looking for a Staff Machine Learning Engineer who will drive the design, development, and deployment of scalable ML and AI solutions across Walmart’s enterprise ecosystem.
You will partner closely with data scientists, data engineers, and application developers to operationalize models, improve performance, and deliver AI-driven capabilities that power automation, decision-making, and intelligent insights at scale.
This is a hands-on technical role that requires deep expertise in machine learning engineering, strong software development skills, and an eagerness to make AI tangible through robust production systems.
What You’ll Do
- End-to-End ML Ownership: Design, develop, and deploy ML models into production systems at scale — from data ingestion to serving.
- ML Platform Development: Build reusable frameworks, tools, and pipelines to accelerate model development and deployment (MLOps).
- Collaboration: Work with data scientists to productionize models, optimize training/inference, and integrate into enterprise applications.
- Architecture & Design: Design scalable ML systems leveraging modern cloud technologies and distributed architectures.
- Performance & Reliability: Drive improvements in model performance, scalability, and monitoring using observability tools.
- Automation: Streamline experimentation, deployment, and retraining processes through CI/CD and automation.
- Cross-Functional Influence: Collaborate with data engineering, software engineering, and platform teams to ensure smooth data and model workflows.
- Innovation: Evaluate and implement emerging AI/ML technologies, including generative AI, LLMs, and agentic AI systems, to solve real-world business challenges.
- Mentorship: Provide technical guidance and mentorship to other ML engineers and data scientists, setting high standards for ML quality and engineering rigor.
What You’ll Bring
- Education: Bachelor’s or Master’s degree in Computer Science, Machine Learning, Data Science, or a related field.
- Experience: 8+ years of hands-on experience in machine learning and software engineering, with at least 2 years in a staff-level or lead technical role.
- Technical Expertise:
- Strong programming skills in Python, Java, or Scala.
- Deep understanding of machine learning algorithms, model optimization, and statistical techniques.
- Proficiency in ML frameworks: TensorFlow, PyTorch, Scikit-learn, or XGBoost.
- Experience with data engineering workflows — Spark, Airflow, Kafka, or similar tools.
- Expertise in MLOps practices (CI/CD for ML, model monitoring, feature stores, ML pipeline orchestration).
- Hands-on experience with cloud ML platforms (AWS Sagemaker, Azure ML, Vertex AI, or Databricks).
- Familiarity with containerization and orchestration (Docker, Kubernetes).
- Strong background in SQL and data manipulation for large-scale datasets.
- AI & GenAI Experience:
- Exposure to LLMs, generative AI, and retrieval-augmented generation (RAG) frameworks.
- Experience fine-tuning, deploying, or integrating LLMs into business applications is a strong plus.
- Soft Skills & Leadership:
- Ability to bridge science and engineering — translating models into scalable production systems.
- Excellent communication and stakeholder management skills.
- Passion for mentoring, knowledge sharing, and driving cross-team collaboration.
- Strong analytical and problem-solving mindset with a bias for action and innovation.
About Walmart Global Tech
Imagine working in an environment where one line of code can make life easier for hundreds of millions of people. Thats what we do at Walmart Global Tech. Were a team of software engineers, data scientists, cybersecurity experts and service professionals within the worlds leading retailer who make an epic impact and are at the forefront of the next retail disruption. People are why we innovate, and people power our innovations. We are people-led and tech-empowered.
We train our team in the skillsets of the future and bring in experts like you to help us grow. We have roles for those chasing their first opportunity as well as those looking for the opportunity that will define their career. Here, you can kickstart a great career in tech, gain new skills and experience for virtually every industry, or leverage your expertise to innovate at scale, impact millions and reimagine the future of retail.
Walmart’s culture sets us apart, and we know being together helps us innovate, learn and grow great careers. This role is based in our [Bangalore/Chennai] office for daily work, with the flexibility for associates to manage their personal lives.
Benefits
Beyond our great compensation package, you can receive incentive awards for your performance. Other great perks include a host of best-in-class benefits maternity and parental leave, PTO, health benefits, and much more.
Belonging
We aim to create a culture where every associate feels valued for who they are, rooted in respect for the individual. Our goal is to foster a sense of belonging, to create opportunities for all our associates, customers and suppliers, and to be a Walmart for everyone.
At Walmart, our vision is "everyone included." By fostering a workplace culture where everyone isand feelsincluded, everyone wins. Our associates and customers reflect the makeup of all 19 countries where we operate. By making Walmart a welcoming place where all people feel like they belong, were able to engage associates, strengthen our business, improve our ability to serve customers, and support the communities where we operate.
Equal Opportunity Employer:
Walmart, Inc. is an Equal Opportunity Employer By Choice. We believe we are best equipped to help our associates, customers and the communities we serve live better when we really know them. That means understanding, respecting, and valuing unique styles, experiences, identities, ideas and opinions while being welcoming of all people.
Minimum Qualifications...
Outlined below are the required minimum qualifications for this position. If none are listed, there are no minimum qualifications.
Option 1: Bachelor's degree in computer science, computer engineering, computer information systems, software engineering, or related area and 4 years’ experience in software engineering, machine learning engineering, AI systems or related area.Option 2: 6 years’ experience in software engineering, machine learning engineering, AI systems or related area.
Preferred Qualifications...
Outlined below are the optional preferred qualifications for this position. If none are listed, there are no preferred qualifications.
Master’s degree in computer science, computer engineering, computer information systems, software engineering, or related area and 2 years' experience in software engineering, machine learning engineering, AI systems or related area., Prior work experience in accessibility best practices., Prior work experience in creating inclusive digital experiences, demonstrating knowledge in implementing Web Content Accessibility Guidelines (WCAG) 2.2 AA standards, assistive technologies, and integrating digital accessibility seamlessly.Primary Location...
Pardhanani Wilshire Ii, Cessna Business Park, Kadubeesanahalli Village, Varthur Hobli , IndiaWalmart
103 jobs posted
About the job
Similar Jobs
Discover more opportunities that match your interests
24 days agoStaff Machine Learning Engineer (Platform)
EarnIn
Mountain View, USView details- 17 days ago
Senior Staff Machine Learning Engineer
Patreon
San FranciscoView details
27 days agoMachine Learning Engineer
EarnIn
Bengaluru, IndiaView details- 27 days ago
Staff Machine Learning Engineer
Quantcast
View details - 20 days ago
Staff Machine Learning Engineer
Bumble
View details - 12 days ago
Machine Learning Engineer
Twilio
RemoteView details - 20 days ago
Staff Machine Learning Engineer, AI Translation
Airbnb
United StatesView details - 18 days ago
Staff Machine Learning Engineer, Content Recommendations
Pinterest
RemoteView details
9 days agoStaff Machine Learning Engineer (Applied ML)
EarnIn
Mountain View, USView details- 7 days ago
[CPLB] Staff Machine Learning Engineer (PMO)
Coupang
Seoul, South KoreaView details
Looking for something different?
Browse all AI jobs