Staff, Software Engineer - Machine Learning Platform

$143,000 - $286,000/Yr

Walmart - Sunnyvale, CA

Part-time,Full-time - Entry Level

Sunnyvale, CA

General Merchandise Retailers

About the position

The position involves developing and deploying foundational AI layers and Gen AI models for online inferencing, crucial for applications like Ads moderation and social commerce. The role focuses on creating a robust machine learning platform, maintaining web-based dashboard systems for data insights, and managing big-data pipelines. The candidate will also handle public cloud services to enhance storage and computation capabilities on the Google Cloud Platform.

Responsibilities

• Develop and deploy foundational AI layers and Gen AI models for online inferencing.
• Maintain a robust, scalable machine learning platform for end-to-end operations.
• Create and maintain web-based dashboard systems for visual insights and model performance evaluation.
• Develop, test, and deploy big-data and machine learning pipelines.
• Manage public cloud services to scale storage and computation capabilities.

Requirements

• Proficiency in at least one programming language such as Python, Java, C++, or JavaScript.
• Solid understanding of data structures and algorithms.
• Extensive knowledge in distributed inferencing in GPUs, LLM serving framework, and PyTorch/Tensorflow.
• Experience with Linux, public cloud computing, Docker, and Kubernetes.
• Experience in developing REST API services.

Nice-to-haves

• Master's degree in Computer Science, Computer Engineering, Computer Information Systems, Software Engineering, or related area with 2 years' experience in software engineering or related area.
• Background in creating inclusive digital experiences and knowledge of WCAG 2.2 AA standards.
• Knowledge of accessibility best practices.

Benefits

• 401(k) match
• Stock purchase plan
• Paid maternity and parental leave
• PTO
• Multiple health plans
• Short-term and long-term disability
• Company discounts
• Military Leave Pay
• Adoption and surrogacy expense reimbursement
• Live Better U education benefit program