Leads the design, development, deployment, and operation of scalable machine learning systems, with emphasis on automated research capabilities for model training, optimization, experimentation, and evaluation. Builds reliable ML services, APIs, and data pipelines; applies MLOps practices such as CI/CD, model versioning, monitoring, and retraining; ensures governance and security compliance; and mentors cross-functional engineering and data science teams.
Position Summary...The Senior Machine Learning Engineer will lead the design, development, and deployment of scalable machine learning solutions that address complex business challenges, with a primary focus on building auto-research capabilities that automate model training optimization, experimentation, and evaluation to accelerate the ML development lifecycle. This role requires collaboration with cross-functional teams to deliver robust, production-ready models and data pipelines while ensuring compliance with security and governance standards. Additionally, they will provide technical leadership, mentor peers, and drive continuous improvement to enhance system reliability and innovation aligned with organizational goals.
About the team:
Our team develops foundational models tailored for Walmart’s e-commerce platform by leveraging large-scale user interactions to create rich representations of users, products, and search queries. These models enable scalable search, advertising, recommendation, and personalization solutions. Team members apply expertise in data science, machine learning, and software engineering to optimize model performance, training, and deployment. With a strong focus on data quality, advanced analytics, and impactful visualization, the team delivers actionable insights that influence key business outcomes. Cross-functional collaboration drives innovation and continuous improvement in a dynamic technology environment.What you'll do...
What you'll bring:
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
Option 2: 5 years’ experience in software engineering, machine learning engineering, AI systems or related area.Preferred Qualifications...
About the team:
Our team develops foundational models tailored for Walmart’s e-commerce platform by leveraging large-scale user interactions to create rich representations of users, products, and search queries. These models enable scalable search, advertising, recommendation, and personalization solutions. Team members apply expertise in data science, machine learning, and software engineering to optimize model performance, training, and deployment. With a strong focus on data quality, advanced analytics, and impactful visualization, the team delivers actionable insights that influence key business outcomes. Cross-functional collaboration drives innovation and continuous improvement in a dynamic technology environment.What you'll do...
- Collaborate with engineers, data scientists, and product teams to deliver machine learning features and models.
- Design and develop auto-research capabilities that automate model training, optimization, experimentation, and evaluation, enabling AI agents to iteratively tune hyperparameters and run experiments against defined objective metrics.
- Surface the best-performing models with minimal manual intervention, accelerating the overall machine learning development lifecycle.
- Implement, test, and support machine learning services, APIs, and data pipelines, ensuring quality and reliability.
- Apply MLOps practices, including CI/CD, model versioning, and automated retraining, to sustain performance.
- Conduct rigorous model evaluation, monitoring, and maintenance to address model drift and performance issues.
- Ensure secure, compliant machine learning engineering aligned with enterprise AI governance standards.
What you'll bring:
- Proven expertise in designing, developing, and deploying scalable machine learning systems using frameworks such as TensorFlow, PyTorch, and Scikit-learn.
- Strong proficiency in programming languages including Python, SQL, Java, and C++ to build production-ready machine learning applications.
- Hands-on experience operating agentic or autonomous research systems.
- Experience running automated experimentation against defined objective metrics through AutoResearch loops.
- Experience with machine learning lifecycle management, including data sourcing, feature engineering, model training, evaluation, deployment, and monitoring.
- Knowledge of MLOps practices such as CI/CD, model versioning, automated retraining, and telemetry to ensure system reliability.
- Ability to implement secure, compliant machine learning pipelines that adhere to enterprise AI governance standards.
- Demonstrated collaboration and mentorship capabilities within cross-functional teams to drive technical excellence.
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
ㅤ
ㅤ
ㅤ
ㅤ
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 3 years’ experience in software engineering, machine learning engineering, AI systems or related area.Option 2: 5 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 1 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...815 Eleventh Ave, Sunnyvale, CA 94089-4731, United States of AmericaWalmart and its subsidiaries are committed to maintaining a drug-free workplace and has a no tolerance policy regarding the use of illegal drugs and alcohol on the job. This policy applies to all employees and aims to create a safe and productive work environment.Walmart Global Tech Sunnyvale, California, USA Office
840 W California Ave, Sunnyvale, CA, United States, 94086
Similar Jobs
Software
Build and optimize high-volume, real-time inference systems for speech and language models. Responsibilities include improving model serving with quantization, batching, caching, attention, and decoding techniques; developing high-performance GPU systems; scaling distributed inference across Kubernetes and multi-node environments; profiling and optimizing code; containerizing models; and ensuring reliable production operation.
Top Skills:
C++Continuous BatchingCudaDistillationKubernetesLlm ServingMulti-GpuMulti-Node InferenceNvidia GpusPaged AttentionPythonQuantizationRayRustSpeculative DecodingTrt-LlmVllm
Software
This role involves optimizing inference performance, managing high-performance systems, scaling models, and full-cycle ownership from research to production in a fast-paced environment.
Top Skills:
C++CudaKubernetesPythonRayRustTrt-LlmVllm
Information Technology • Productivity • Software
As a founding Machine Learning Engineer, you will improve core models, optimize LLM inference, develop data pipelines, and ship features directly to users.
Top Skills:
Chrome ExtensionCloudflare WorkersElectron AppLlmsNode.jsReactTransformer ArchitecturesTypescriptVector Databases
What you need to know about the San Francisco Tech Scene
San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine


