Staff Software Engineer, Snorkel AI
Compute Platform
3+ days a week in office
Data-centric enterprise AI platform for
Be an early applicant
Data-centric enterprise AI platform for
101-200 employees
Be an early applicant
3+ days a week in office
101-200 employees
Company mission
To empower everyone to solve their most impactful problems through data-centric AI.
Role
Who you are
- Bachelor’s or Master’s degree in Computer Science, Software Engineering, or a related field
- 8+ years of experience in backend or infrastructure engineering, with significant expertise in MLOps, and SDK development for AI applications
- Proven track record of architecting and deploying scalable infrastructure for distributed systems, preferably with tools like Prefect, Ray, or Airflow
- Strong expertise in Python and backend frameworks (e.g., FastAPI, Flask)
- Extensive experience with CI/CD pipelines, containerization (Docker), and orchestration (Kubernetes)
- In-depth knowledge of distributed computing, resource scheduling, and fault-tolerant system design
- Exceptional problem-solving skills, with the ability to design solutions for complex compute challenges
- Knowledge of AI/ML workflows
Desirable
- Hands-on experience with vector databases, data processing frameworks, and data connectors for enterprise systems
- Familiarity with monitoring and observability tools like Prometheus, Grafana, or DataDog
- Experience in scaling SDKs and APIs for large user bases
- Strong understanding of distributed system design patterns and performance optimization
- Understanding of the latest AI models and pipeline needs
What the job involves
- As a Staff Compute Platform Engineer, you will be a technical leader shaping the core infrastructure that powers SnorkelFlow’s compute capabilities
- Your expertise in orchestration, MLOps, SDK maintainability, and infrastructure components like data connectors and distributed systems (e.g., Ray) will drive scalable and reliable solutions for the platform
- In this role, you will lead initiatives, mentor engineers, and collaborate across teams to deliver innovative solutions that support SnorkelFlow's mission to simplify AI development
- Architect and lead the development of the compute platform, focusing on scalability, reliability, and performance
- Drive best practices in MLOps and orchestration, ensuring seamless integration with AI and Data pipelines and workflows
- Work with other technical leads to define the technical roadmap for the compute layer, aligning with SnorkelFlow's broader platform strategy
- Identify opportunities to optimize compute workflows and implement solutions that reduce complexity while increasing efficiency
- Design and implement robust orchestration workflows using tools like Prefect and Ray to support distributed and parallel computing
- Build and scale infrastructure for integrating with data sources such as S3, Snowflake, and Databricks
- Ensure fault-tolerant, high-availability compute systems that can handle diverse workloads efficiently
- Oversee the design and development of the SnorkelFlow SDK, ensuring it is intuitive, extensible, and aligned with user needs
- Collaborate with cross-functional teams to expose compute platform functionalities through APIs and SDK interfaces
- Drive improvements in SDK maintainability, including versioning, documentation, and developer tooling
- Partner with AI and Data Platform teams to ensure seamless interoperability between compute workflows and other platform layers
- Work closely with the Application team to design APIs that enable advanced user workflows and orchestration capabilities
- Act as a technical advisor to other teams, providing expertise in compute infrastructure and orchestration
- Define and implement observability strategies, including monitoring tools, dashboards, and logging frameworks, to track compute platform performance
- Optimize system performance by identifying bottlenecks and implementing efficient compute and data processing solutions
- Establish metrics and reporting to ensure continuous improvement of compute workflows
- Mentor engineers across all levels, fostering technical growth and a culture of excellence
- Lead code and design reviews, ensuring adherence to high-quality engineering standards
- Advocate for innovative solutions and inspire the team to push technical boundaries
Our take
Demand for AI is increasing in almost every industry, but the machine learning that fuels it is tedious to set up. In order to train machine learning algorithms, training data must be labelled manually, and this contributes to expenses in time, money and resources when integrating AI within a business.
Snorkel AI develops programmatic approaches to data labelling in order to automate the setup process of machine learning and decrease the required timeframe for a business to begin providing AI services. In addition to these automated data categorization services, the company also integrates data training, management, and analysis into a unified AI-deployment platform.
The complex computer science behind AI setup has resulted in many businesses without computer science backgrounds being left behind, and Snorkel AI has set this as their target audience. As of 2023, this customer base has tripled in size since Snorkel AI's founding and in 2024, enjoyed a new round of funding. With these promising developments, the company continues to innovate the sector.
Kirsty
Company Specialist at Welcome to the Jungle
Insights
Few candidates hear
back within 2 weeks
-13% employee growth in 12 months
Company
Funding (last 2 of 3 rounds)
Aug 2021
$85m
SERIES C
Apr 2021
$35m
SERIES B
Total funding: $135m
Company benefits
- Home office allowance
- Yearly wellness stipend
- Healthy meals, snacks & drinks in office
- 401k with a 100% match up to 5% of annual salary
- Regular team events
- Generous paid parental leave
- Work from home opportunities
- Health insurance
Company values
- We cultivate autonomy across the entire team by being open about our goals, wins, and challenges.
- We get to answers fast, focusing on what works—not what's fancy.
- In our field of AI and software engineering, we believe that diverse thinkers increase collective insights and knowledge.
Company HQ
Centennial, Redwood City, CA
Articles
Leadership
Alex Ratner
(Co-founder & CEO)Studied at Harvard and received a PhD from Stanford. They are also an Assistant Professor at the University of Washington.
Paroma Varma
(Co-founder)They hold a PhD from Stanford and was VP of Corporate Relations at the Society of Women Engineers. They have collaborated with teams at Facebook and Intel.
Braden Hancock
(Co-founder & Head of Technology)They have a PhD in Computer Science from Stanford, and worked as a Research Assistant with the Air Force and MIT. They were also a Software Engineer at Google, and a Researcher at Facebook.
Henry Ehrenberg
(Co-Founder)They studied at Yale and Stanford and worked at Facebook as a Quantitative Engineering Intern and a Senior Applied Research Scientist.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 42 more jobs at Snorkel AI