Staff, Software Engineer at Walmart

Sunnyvale, California, United States

Walmart Logo
Not SpecifiedCompensation
Senior (5 to 8 years), Expert & Leadership (9+ years)Experience Level
Full TimeJob Type
UnknownVisa
Retail, TechnologyIndustries

Requirements

  • Overall 8+ years of experience in machine learning / software engineering, with 2+ years in Generative AI or LLMs
  • Strong experience with Agentic Frameworks, RAG frameworks, vector search technologies and knowledge grounding techniques
  • Expertise in Python, ML frameworks (PyTorch, TensorFlow), and libraries like Hugging Face Transformers
  • Strong backend Programming experience in Python / Golang or Java
  • Proficiency in distributed computing, cloud platforms (GCP, Azure), and scalable architectures
  • Clear Communication, including the ability to build design gaps
  • Ability to multitask between several different requirements and features concurrently
  • Deep knowledge of open-source projects or publications in Generative AI, RAG, or agentic frameworks
  • Ability to perform in a team environment and have a true passion about Technology
  • Provide deeper Insights on Containerization/Docker and experience in CI/CD and deployment pipelines

Responsibilities

  • Design, Develop, fine-tune, and deploy Generative AI models with a focus on RAG frameworks to enhance information retrieval and generation
  • Develop and optimize LLM-based agents for tasks like knowledge grounding, multi-step reasoning, and decision-making
  • Integrate external data sources (e.g., vector databases) with LLMs to build dynamic, context-aware systems
  • Develop scalable, distributed AI systems with a focus on performance, fault tolerance, and disaster recovery
  • Develop detailed design specifications, including modular, extensible, and functional component
  • Develop MVP's into Shippable product to validate requirements and refine designs iteratively
  • Work cross-functionally to design solutions that address both functional and non-functional requirements
  • Establish best practices for building and maintaining AI-driven systems
  • Drive technical direction for large-scale AI initiatives and mentor junior engineers
  • Independently handle high impact, critical software/systems monitoring issues, troubleshoot business and production issues
  • Work closely with Engineering teams, Architects, Product group to provide highly recommended Developer Experience and Gen AI solutions

Skills

Generative AI
LLMs
Agentic Frameworks
RAG
AI Systems
DevOps
Cloud Computing
Infrastructure
Software Engineering
Model Fine-tuning
Scalable Architecture
Production Troubleshooting

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