Senior, Software Engineer - Gen AI at Walmart

Sunnyvale, California, United States

Walmart Logo
$150,000 – $220,000Compensation
Senior (5 to 8 years)Experience Level
InternshipJob Type
UnknownVisa
Artificial Intelligence, AI & Machine LearningIndustries

Requirements

Candidates should possess 6+ years of experience in machine learning engineering, with at least 2+ years dedicated to Generative AI and NLP, demonstrating proficiency in Python, ML frameworks such as PyTorch and TensorFlow, and NLP libraries like Hugging Face Transformers. They should also have hands-on experience with RAG frameworks, vector search tools, and knowledge grounding techniques, along with familiarity with distributed systems, scalable architectures, and cloud platforms like GCP or Azure, and a strong understanding of software design patterns and non-functional requirements.

Responsibilities

The Senior Machine Learning Engineer will design and implement Generative AI solutions utilizing RAG for summarization and knowledge retrieval, build and integrate LLM-based agents leveraging vector search technologies for dynamic responses, fine-tune and optimize models for specific business needs, contribute to building scalable, distributed systems, translate high-level designs into detailed solutions, participate in code reviews, drive technical direction for large-scale AI initiatives, mentor junior engineers, and establish best practices for building and maintaining AI-driven systems.

Skills

Python
PyTorch
TensorFlow
Hugging Face Transformers
RAG frameworks
Vector search tools
Knowledge grounding techniques
Distributed systems
GCP
Azure
Software design patterns
LLM-based agents

Walmart

Global retail chain offering affordable products

About Walmart

Walmart operates a chain of hypermarkets, discount department stores, and grocery stores, providing a wide variety of products at low prices to everyday consumers, families, and communities. The company combines physical retail locations with an e-commerce platform, allowing customers to shop for groceries, apparel, electronics, and household items both in-store and online. Walmart's efficient supply chain management and large-scale purchasing enable it to maintain affordability, setting it apart from competitors. The company also offers financial services and health and wellness services, including pharmacies and vision centers. Walmart's goal is to save people money and support communities, as demonstrated by its efforts during the COVID-19 pandemic and its commitment to social responsibility, including programs for veterans and military families.

Bentonville, ArkansasHeadquarters
1962Year Founded
$5,515MTotal Funding
IPOCompany Stage
Financial Services, Consumer GoodsIndustries
10,001+Employees

Benefits

PTO: Paid vacation, sick time, personal time and holiday time
10% discount on regularly priced general merchandise and fresh fruits and vegetables
6% 401(k) match to all employees, including hourly workers, after one year
Roth IRA available
Associate Stock Purchase Plan
maximum and eligible preventive care covered at 100%
Health reimbursement plans

Risks

Rising U.S. labor costs could pressure profit margins.
Amazon's grocery delivery expansion threatens Walmart's market share.
Fintech investments expose Walmart to financial regulatory risks.

Differentiation

Walmart's omnichannel strategy integrates physical and online retail seamlessly.
The company offers a wide range of products at consistently low prices.
Walmart's commitment to community support enhances its brand reputation.

Upsides

Walmart's investment in fintech expands its financial services offerings.
Robotics integration in distribution centers boosts operational efficiency.
Culturally themed product lines attract diverse customer segments.

Land your dream remote job 3x faster with AI