Software Engineer — AI/ML

As a Senior AI/ML Engineer at Snorkel AI, you will be responsible for building systems to support large-scale machine learning and foundation model workloads. Your main responsibilities will include owning the architecture, design, development, and operations of AI/ML systems, integrating the latest foundation model technologies, prototyping and maintaining scalable back-end services, designing interfaces between internal services, and being an engaged team player in a customer-focused environment. With a focus on delivering distributed and ML systems in a production setting, you should have experience with distributed compute frameworks, deep learning frameworks, and designing efficient scalable data storage systems. This role offers a salary range of $191,000.00 - $225,000.00 in Tier 1 locations.

• Own the architecture, design, development, and operations of large-scale systems designed for AI/ML tasks including distributed compute systems, data management systems, data engineering workflow systems, and end user experiences
• Recognize and act on opportunities to integrate the latest foundation model and related technologies to power user workflows
• Prototype, optimize, and maintain scalable back-end services that will power new ML and foundation model development workflows
• Design extensible and testable interfaces between internal services including the underlying storage and data models
• Be an engaged team player in a customer-focused cross-functional environment where you will feel excited to take on whatever is most impactful for the company and product
• Work a hybrid schedule with one or two days per week in our Redwood City HQ and work remotely with "No Meeting" Tuesdays and Thursdays

• 4+ years experience in delivering distributed and ML systems and services in a production setting for cloud-native applications
• Experience with distributed compute frameworks and deep learning frameworks
• Ability to design and build efficient scalable data storage, compute, and retrieval systems for AI/ML tasks
• Strong communication and coding skills with emphasis on designing for scale and robustness
• Experience owning the delivery of large multi-person projects
• 8+ years of professional software engineering experience (preferred)
• Experience with architecting and developing production web-scale systems (monitoring, telemetry, performance, reliability, triage and debug) (preferred)
• Strong development and debugging skills in Python (preferred)
• Experience working with foundation models (e.g. large language models) (preferred)
• Experience developing enterprise software products for machine learning and/or data science applications (preferred)

• Comprehensive medical, dental, and vision plans for Snorkelers and their families
• Yearly wellness stipend
• 401k program for future planning
• Parental leave program with up to 20 weeks of paid time off
• Workstation setup allowance
• Equal Employment Opportunity employer
• Commitment to building a diverse team
• Prohibition of discrimination and harassment based on various characteristics
• Reasonable accommodation for individuals with disabilities during the application process and employment