STAFF, SOFTWARE ENGINEER (Machine Learning)

Walmart; Bengaluru; On-site; INR 4,000,000 - 6,000,000; Full time

Benefits offered by this job

Maternity/Parental leave

Health benefits

Paid time off

Job summary

Walmart Global Tech is seeking a Staff Software Engineer (MLE) to design and deploy low-latency computer vision and GenAI pipelines at scale from Bangalore/Chennai. You will collaborate with data scientists and engineers to deliver ML solutions that power critical business processes and improve user experiences for associates and customers.

You will drive end-to-end productionization of models, ensure performance, reliability, and observability, and mentor teams across data science and software

Qualifications

• Bachelor's degree in computer science, computer engineering, or a related area.
• 6+ years of software engineering or related experience may substitute.
• Experience running large-scale ML workloads on GPUs is required or highly desirable.
• Strong hands-on experience with Nvidia DeepStream and Triton Inference Server is expected.

Responsibilities

• Build and maintain low-latency, high-throughput computer vision pipelines at Walmart scale.
• Deploy end-to-end pipelines for ML and GenAI solutions in production.
• Drive data-driven insights by supporting ML models and algorithms across divisions.
• Lead data gathering, validation, and synthesis to support analytics projects.
• Communicate recommendations to partners and influence future plans.
• Own automation with vision system design, API interfaces, and SQL/NoSQL concepts.
• Enhance performance and address bottlenecks in web apps and ML models.

Skills

GPU ML workloads

NVIDIA DeepStream

NVIDIA Triton Inference Server

PyTorch in production

GStreamer

CUDA C++

Kubernetes

ML observability

Python

REST API development

Education

Bachelor's degree in computer science or related field

Tools

NVIDIA DeepStream

NVIDIA Triton

GStreamer

TensorRT

PyTorch

Docker

Kubernetes

Job description

Position Summary...

Skill: Machine Learning Engineer

Exp range: 8 - 13 years

About Team:

This position is in the data science & engineering team called Global Governance under the Corporate Technology Services organisation. Enterprise areas of Walmart like Finance, Indirect Procurement, People Tech and others are in the midst of massive digital transformation.

Our team is building products for these areas from scratch, dealing with challenges from gathering user data, developing machine learning models to understand business processes better and making them more holistically data driven and intelligent. Video Analytics, Agentic Workflows, Object Detection and Segmentation, Optimisation, Anomaly Detection are only some of the areas we are actively working on. We leverage cutting edge deep learning and computer vision, GenAI technologies to solve business critical problems.

Do you enjoy developing, deploying and handling complex production environments and collaborating with cross-functional teams to maintain optimal application performance? As a Staff Software Engineer, you are responsible for translating requirements into technical solutions with the focus of creating the best user experience for our associates and customers. You will be part of a project team consisting of other data scientists and engineers who will work together to deliver & deploy ML and GenAI models at scale, business value and ensure code quality. Communication is also key as you will need to work closely with product owners, data scientists & engineers.

What you’ll do:

As a Staff Software Engineer (MLE) for Walmart, you’ll have the opportunity to:

• Build and maintain low-latency/high-throughput Computer Vision (real-time/batch) pipelines at Walmart scale and complexity.
• Build scalable and reusable components and productionize the solution by deploying end-to-end pipelines for ML & GenAI solutions.
• Drive data-derived insights across a wide range of divisions by supporting statistical models, machine learning algorithms, and computational algorithms based on business initiatives.
• Direct the gathering of data, assessing data validity, and synthesizing data into large analytics datasets to support project goals.
• Communicate recommendations to business partners and influence future plans based on insights.
• Take ownership of achieving complete automation with design for the Vision solution, with a good level of comfort with shell, API design, queuing, and advanced SQL/NoSQL concepts.
• Utilize big data analytics and advanced MLOps data science techniques to identify trends, patterns, and discrepancies in data.
• Enhance the performance of applications while identifying and addressing bottlenecks from the standpoint of both web applications and machine learning models.
• Apply your computer vision and forecasting expertise where required.

What you’ll bring:

• Good understanding of running large-scale ML workloads on GPUs.
• Strong hands‑on experience with Nvidia DeepStream — pipeline configuration, plugin authoring, multi‑stream multiplexing, and production‑level debugging of inference graphs.
• Strong hands‑on experience with Nvidia Triton Inference Server — model repository layout, ensemble pipelines, dynamic batching, and TensorRT/PyTorch/ONNX backend configuration.
• Experience deploying and scaling PyTorch models in production — TorchScript, TensorRT export, FP16/INT8 precision tuning, and model versioning.
• Good expertise in GStreamer — element graph design, message handling, pad linking, and end‑to‑end pipeline latency profiling from source to downstream integration(s).
• Good to have working knowledge of C++ for CUDA and DeepStream/GStreamer plugin development
• Good hands‑on experience building ML observability stacks for complex ML and GenAI pipelines — metrics, monitoring, alerting, and dashboarding (e.g., Grafana/Prometheus with PromQL, latency‑SLO instrumentation; log aggregation via Splunk), drift and model/data-quality detection.
• Strong Kubernetes operations skills — pod lifecycle management, GPU scheduling, custom Helm charts, and safe‑deployment patterns such as canary rollouts and automated rollback.
• Good understanding of Git, MLOps, and CI/CD automation (model/data/code versioning).
• Good understanding of networking fundamentals, including IP addressing & network security.
• Strong Python proficiency — pipeline scripting, test‑harness development, automation tooling, and data‑processing utilities; multiprocessing (and async/threading for I/O‑bound work) for performance optimisation; and NumPy/Pandas for data preprocessing and feature engineering.
• Experience designing and running testing frameworks across workloads: E2E and regression test suites, pre-deployment validation, and safe rollout strategies (e.g., silent/shadow testing), applying testing best practices to catch regressions before production.
• Knowledge of cloud technologies such as GCP or Azure, and knowledge of RDBMS and NoSQL/caching solutions.
• Working understanding of object detection, object segmentation, and tracking models and their challenges.
• Exposure to GenAI deployment patterns — LLM inference serving and model optimization for production environments.
• Good to have working knowledge of agentic workflows, RAG pipelines, MCPs/Skills, and VectorDB (e.g., Milvus/FAISS).
• Good understanding of authentication, authorization/permissions, and privacy best practices in web services.
• Basic understanding of developing REST APIs using Node.js & Spring Boot.
• High‑level proficiency in using the following technologies to build modern web applications: JavaScript, TypeScript, React, React Native, HTML, and CSS/SASS.
• Comfortable with ambiguity; a self‑motivated learner and builder.

About Walmart Global Tech

Imagine working in an environment where one line of code can make life easier for hundreds of millions of people. That’s what we do at Walmart Global Tech. We’re a team of software engineers, data scientists, cybersecurity expert's and service professionals within the world’s 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 is—and feels—included, 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, we’re 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 Opportunities 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 inclusive of all people.

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 or related area.
• Option 2: 6 years’ experience in software engineering or related area.

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 or related area.

Primary Location

Block- 1, Prestige Tech Pacific Park, Kadubeesanahalli Village, Varthur Hobli , India