Join to apply for the

Staff Software Engineer — AI Platform

role at

Snorkel AI Join to apply for the

Staff Software Engineer — AI Platform

role at

Snorkel AI Get AI-powered advice on this job and more exclusive features. About Snorkel

At Snorkel, we believe meaningful AI doesn’t start with the model, it starts with the data. About Snorkel

At Snorkel, we believe meaningful AI doesn’t start with the model, it starts with the data.

We’re on a mission to help enterprises transform expert knowledge into specialized AI at scale. The AI landscape has gone through incredible changes between 2015, when Snorkel started as a research project in the Stanford AI Lab, to the generative AI breakthroughs of today. But one thing has remained constant: the data you use to build AI is the key to achieving differentiation, high performance, and production-ready systems. We work with some of the world’s largest organizations to empower scientists, engineers, financial experts, product creators, journalists, and more to build custom AI with their data faster than ever before. Excited to help us redefine how AI is built? Apply to be the newest Snorkeler!

As a

Staff AI Platform Engineer , you will play a pivotal role in designing and implementing the core AI pipelines that drive SnorkelFlow’s AI capabilities. You will act as a technical leader, setting architectural direction, mentoring team members, and collaborating across verticals to ensure the AI Platform aligns with product goals and organizational strategy.

This role requires hands-on experience with

large language models (LLMs) , requiring deep expertise in AI development, especially in designing and deploying systems that leverage LLMs and more traditional NLP solutions for production workflows.

Main Responsibilities

Technical Leadership

Provide technical direction for the design and development of AI pipelines, ensuring scalability, robustness, and extensibility. Serve as a mentor and guide for engineers on the AI Platform team, fostering growth and technical excellence. Identify and drive high-impact projects aligned with business objectives and product needs.

AI Pipeline Development

Architect, design, and maintain AI pipelines for labeling, embeddings, training, and deploying models into production. Lead the development and optimization of MLFlow pipelines for deployment. Build and deploy foundational models that serve as the backbone for SnorkelFlow’s core product capabilities. Partner with the Compute Platform team to ensure seamless integration with orchestration tools and infrastructure.

LLM Tooling Development

Develop and deploy LLM-based systems for production workflows, focusing on efficiency, scalability, and reproducibility. Create AI training framework pipelines that will allow LLM usage in applications, including fine-tuning, pruning, distillation, and foundational model training. Integrate APIs from providers such as OpenAI, Anthropic, and Gemini into SnorkelFlow’s pipelines.

Backend Development

Oversee the integration of backend services for managing LLM calls and API interactions. Collaborate with the Data Platform team to define data requirements and ensure smooth interoperability. Work with the Application team to design and implement APIs that power application workflows.

Observability and Monitoring

Establish observability standards for AI pipelines, including tools and dashboards for monitoring model performance and debugging. Define key metrics for system health and optimization.

Cross-Functional Collaboration

Act as a thought leader, collaborating with Data Platform, Compute Platform, Application, Product and other internal teams to deliver cohesive, scalable solutions. Partner with stakeholders to translate product goals into technical roadmaps and

Required Qualifications

Bachelor’s or Master’s degree in Computer Science, Machine Learning, or a related field. 8+ years of experience with a good portion of them in AI development, including hands-on work with AI in production systems. Strong expertise in Python and deep learning frameworks such as PyTorch. Proven experience leading technical projects and mentoring engineers. Proficiency with CI/CD pipelines for machine learning workflows. Deep understanding of LLM architectures, fine-tuning, and deployment methodologies. Strong communication skills, with an emphasis on scalable and reliable system design.

Preferred Qualifications

Expertise in NLP and familiarity with libraries such as Hugging Face Transformers, spaCy, scikit-learn, or XGBoost. Familiarity with multimodal AI concepts, including vision and audio tasks. Experience working with APIs and foundational model providers such as OpenAI, Anthropic, or Gemini. Knowledge of MLOps tools and practices, such as MLflow, Kubernetes, or Ray. Experience building APIs or SDKs for AI services.

What We Offer

A leadership role with the opportunity to influence the technical direction of SnorkelFlow. Competitive salary and benefits tailored to your experience. Hybrid work environment with 3 days per week at our Redwood City HQ and SF Office. "No Meeting" Tuesdays and Thursdays to focus on deep work. The chance to work on cutting-edge infrastructure and drive impactful change in an innovative, fast-paced environment.

Be Your Best at Snorkel

Joining Snorkel AI means becoming part of a company that has market proven solutions, robust funding, and is scaling rapidly—offering a unique combination of stability and the excitement of high growth. As a member of our team, you’ll have meaningful opportunities to shape priorities and initiatives, influence key strategic decisions, and directly impact our ongoing success. Whether you’re looking to deepen your technical expertise, explore leadership opportunities, or learn new skills across multiple functions, you’re fully supported in building your career in an environment designed for growth, learning, and shared success.

Snorkel AI is proud to be an Equal Employment Opportunity employer and is committed to building a team that represents a variety of backgrounds, perspectives, and skills. Snorkel AI embraces diversity and provides equal employment opportunities to all employees and applicants for employment. Snorkel AI prohibits discrimination and harassment of any type on the basis of race, color, religion, age, sex, national origin, disability status, genetics, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state, or local law. All employment is decided on the basis of qualifications, performance, merit, and business need.

We will ensure that individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process, to perform essential job functions, and to receive other benefits and privileges of employment. Please contact us to request accommodation. Seniority level

Seniority level Mid-Senior level Employment type

Employment type Full-time Job function

Job function Engineering and Information Technology Industries Software Development Referrals increase your chances of interviewing at Snorkel AI by 2x Get notified about new Staff Software Engineer jobs in

San Francisco, CA . Staff Software Engineer - Detection and Response Platform

San Mateo, CA $200,000.00-$300,000.00 1 week ago Sr. Software Engineer - Payments Support

San Francisco, CA $138,400.00-$173,000.00 2 weeks ago Agentic Software Engineer / AI Systems Developer (Contract/Part-Time Lead)

Staff Software Engineer, Frontend (Rapid Prototyping Leader)

San Francisco, CA $202,000.00-$238,000.00 2 weeks ago Staff Software Engineer, Demand Engineering - Slack

Software Engineering Technical Leader (Hybrid Remote- San Jose, CA)

San Francisco, CA $198,600.00-$282,900.00 2 weeks ago Senior Software Engineer, Product Foundations

San Francisco, CA $155,000.00-$215,000.00 13 hours ago Redwood City, CA $140,000.00-$198,000.00 4 days ago Senior Robotics Software Engineer, Planning and Control

San Francisco, CA $150,000.00-$240,000.00 1 month ago Staff Software Engineer, Vessel Performance Modeling

San Francisco, CA $170,000.00-$220,000.00 2 weeks ago Senior/Staff Motor Control Software Engineer

Senior Software Engineer, Robotics Reliability (Hardware in the Loop Experience Required)

San Francisco, CA $214,000.00-$247,500.00 9 months ago San Francisco, CA $120,000.00-$180,000.00 5 months ago San Francisco, CA $200,000.00-$250,000.00 6 days ago San Francisco, CA $125,700.00-$334,600.00 2 weeks ago San Francisco, CA $130,000.00-$200,000.00 9 months ago San Francisco, CA $250,000.00-$300,000.00 4 months ago San Francisco, CA $100,000.00-$270,000.00 1 month ago San Francisco, CA $140,000.00-$190,000.00 9 months ago Software Engineer - Hybrid - San Francisco

Staff Software Engineer, Distributed Systems

San Francisco, CA $194,000.00-$250,000.00 1 month ago San Francisco, CA $160,000.00-$210,000.00 9 months ago San Francisco, CA $160,000.00-$210,000.00 1 month ago Senior Software Engineer - (Full-Stack )

San Francisco, CA $150,000.00-$300,000.00 1 month ago We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.

#J-18808-Ljbffr