Distinguished, Software Engineer - Machine Learning

Reference: R-2629499

Posted:

Location:

SUNNYVALE, CA, United States; BELLEVUE, WASalary: $169000 - $338000/year

Type:
Full time / Regular/Permanent Company:
Walmart Business Segment:
Home Office Role summary:

As a Distinguished Software Engineer, you will lead the design and development of scalable, cloud-native AI-powered systems, ensuring architectural integrity and operational excellence. This role requires translating strategic objectives into robust technical solutions while fostering collaboration across engineering teams. You will champion best practices in software development, continuous integration, and reliability engineering, driving innovation and continuous improvement. By providing technical leadership and mentorship, you will enhance system performance, scalability, and maintainability, contributing to the achievement of business goals with a strong commitment to quality and operational readiness.

About the team:

The AdTech Measurement and Reporting team develops scalable, cloud-native platforms for real-time data processing and AI integration. This team delivers precise performance insights that support strategic decision-making and business goals. Collaborating across functions, members design robust tools to enhance reporting accuracy and efficiency. By applying advanced engineering principles and data-driven methods, the team ensures reliable, high-quality outputs. Focused on continuous innovation and alignment with technical strategy, this group advances Walmart’s technology solutions through disciplined development, testing, and delivery practices, emphasizing operational excellence and scalability.

What you'll do:
Solve world class scale AdTech problems using ML (e.g., transformer model) and statistical methods together with a group talented engineers with doctoral degrees and industry experiences in top-tier tech companies. Lead platform development for a suite of measurement solutions from 0 to 1— including incrementality, attribution, data cleanroom measurement, partner integrations, and GenAI-powered reporting — supporting both real-time optimization and offline reporting with the absolute stability and correctness demanded by Walmart's multi-billion-dollar digital advertising business.

Define technical direction and conduct design reviews for engineering teams while partnering with engineering leadership and cross-functional stakeholders to align technical strategy with business goals. Troubleshoot business and production issues by gathering information (issue, impact, criticality, possible root cause), engage support teams to assist in resolution of issues, formulate an action plan, performing actions as designated in plan, interpret the results to determine further action, and complete online documentation.

What you'll bring: 10+ years of experience in ML and data intensive platform development. 4+ years of experience as staff+ level technical lead. Familiar with distributed data processing tools and databases, e.g., Spark streaming, Flink, Kafka, RAY, Big Query, Druid, Casandra, etc. Skilled at leveraging AI to enhance productivity and drive innovation. Familiar with GenAI skills and techniques e.g., MCP, RAG, Re-ranking, Agent Skill, Progressive disclosure, Context engineering, Agent-to-agent protocol, fine-tuning, etc.

Familiar with prevailing ML libraries e.g., pyspark, pytorch, CUDA, Triton etc. Experience implementing statistical and machine learning methods for data classification, regression, ranking, and/or recommendation. Experience implementing transformer-based model training and real-time inference on GPU backed infrastructure. Experience working cross-functionally with data scientists, machine learning engineers, and product managers.…