(USA) Senior, Machine Learning Engineer

Role summary:

The Senior Machine Learning Engineer will lead the design, development, and deployment of scalable machine learning solutions that address complex business challenges. This role involves collaborating closely with cross-functional teams to deliver robust models, ensuring high-quality code, and maintaining reliable ML pipelines. The position requires expertise in model evaluation, MLOps practices, and secure AI engineering. The successful candidate will drive continuous improvement, mentor peers, and uphold technical excellence while aligning solutions with organizational goals and compliance standards. This role demands strong problem-solving skills and a commitment to innovation and operational resilience.

About the team:

The eCommerce Search team manages the full technology stack enabling product search and discovery across Walmart's global eCommerce platforms. Supported by microservices, extensive data and feature pipelines, search engines, and ML-serving infrastructure, the team processes millions of daily queries. Key focus areas include query and intent understanding using NLP and deep learning, real-time autocomplete, complex retrieval methods, multi-phase ML ranking models, and data pipelines that feed search indices and analytics. This team's work directly influences how millions of customers locate products efficiently and effectively. Our systems tackle:

• Query & Intent Understanding - product type classification, intent recognition, and sequence tagging (NLP and deep learning)

• Autocomplete - real-time query suggestions at low-latency

• Retrieval - Complex query construction, faceted navigation, semantic and lexical retrieval

• Ranking - multi-phase ML models (LTR/Neural) optimizing for customer satisfaction

• Data & Feature Pipelines - feeding search indices, feature stores, and analytics platforms.

• What you'll do:

• Collaborate with engineers, data scientists, and product teams to deliver machine learning features and models aligned with business objectives.

• Design, develop, and deploy scalable ML systems using appropriate algorithms, frameworks, and coding languages.

• Implement and maintain data pipelines, APIs, and model components ensuring quality, security, and compliance.

• Apply MLOps practices including CI/CD, model versioning, monitoring, and automated retraining to sustain performance.

• Conduct rigorous model evaluation, validation, and performance monitoring to ensure accuracy and reliability.

• Mentor team members on best practices, code quality, and model evaluation techniques.

• Participate in architectural discussions to enhance system scalability and maintainability.

• Build and optimize high-throughput, low-latency services, applying best practices around distributed systems, fault-tolerance, and concurrency

• Run A/B experiments; measuring impact on search quality and business metrics

What you'll bring:

• Proven expertise in designing, developing, and deploying scalable machine learning models using frameworks such as TensorFlow, PyTorch, and Scikit-learn.
• Strong programming skills in Python, SQL, Java, C++, or R, with experience writing clean, production-ready code.
• Proficiency in MLOps practices including CI/CD pipelines, model versioning, automated retraining, and monitoring for performance and drift.
• Experience with hyperparameter optimization, model evaluation, and feature engineering to ensure robust ML solutions.
• Ability to collaborate effectively with cross-functional teams and mentor junior engineers.
• Knowledge of secure, compliant AI engineering and data pipeline development.
• Familiarity with large language modeling and GPU optimization techniques.

At Walmart, we offer competitive pay as well as performance-based bonus awards and other great benefits for a happier mind, body, and wallet. 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. You will also receive PTO and/or PPTO that can be used for vacation, sick leave, holidays, or other purposes. The amount you receive depends on your job classification and length of employment. It will meet or exceed the requirements of paid sick leave laws, where applicable. For information about PTO, see https://one.walmart.com/notices. Live Better U is a Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam''s Club facilities. Programs range from high school completion to bachelor''s degrees, including English Language Learning and short-form certificates. Tuition, books, and fees are completely paid for by Walmart.

Eligibility requirements apply to some benefits and may depend on your job classification and length of employment. Benefits are subject to change and may be subject to a specific plan or program terms.

For information about benefits and eligibility, see One.Walmart.

The annual salary range for this position is $143,000.00 - $286,000.00 Additional compensation includes annual or quarterly performance bonuses. Additional compensation for certain positions may also include :

• Stock