← All roles

Senior / Staff Software Engineer — AI Platform

Snorkel AI· 312 - Engineering
Apply on company site
Hybrid (New York City, NY; San Francisco, CA)full-timePosted 1mo ago

About the role

Senior / Staff Software Engineer role within Snorkel AI's AI Platform team, building infrastructure for large-scale agentic workloads, synthetic data, evaluation, simulation environments, and LLM infrastructure to support non-deterministic AI workloads at scale.

Responsibilities

  • Design and build infrastructure for running large-scale agentic workloads, including multi-step agents interacting with tools, external services, sandboxes, and simulated environments
  • Build scalable synthetic data generation and automated labeling systems for high-quality training/evaluation datasets
  • Design evaluation infrastructure for measuring AI system behavior across models, prompts, tools, environments, and multi-step trajectories; include reproducible experiments and continuous evaluation
  • Build orchestration and distributed compute systems for running thousands to millions of AI experiments and simulations across heterogeneous compute environments
  • Develop infrastructure for agent simulation environments, including provisioning, isolation, lifecycle management, and scalable execution
  • Build and operate LLM infrastructure for routing, rate limiting, retries, caching, provider failover, cost attribution, and efficient execution across model providers
  • Instrument agent and model workloads to capture traces, interactions, environment state, evaluation results, latency, reliability, and cost
  • Design systems to make non-deterministic workloads reproducible and measurable for experiment comparison and regression diagnosis
  • Improve developer experience for AI experimentation with APIs, SDKs, workflow abstractions, and tooling for large-scale production execution
  • Collaborate with research, product, and engineering teams to turn experimental workflows into reliable platform capabilities

Qualifications

  • 5+ years building production software systems with experience in AI/ML infrastructure, ML platforms, distributed systems, data platforms, or backend infrastructure
  • Experience operating non-deterministic AI or ML workloads in production or at significant scale
  • Experience building infrastructure for experimentation, evaluation, model development, synthetic data, agentic workflows, training, inference, or production ML systems
  • Strong proficiency in Python and production-quality APIs, services, and developer tooling
  • Strong background in distributed systems and cloud platforms (AWS preferred) including compute orchestration, storage, networking, isolation, and failure handling
  • Experience with workflow or distributed execution frameworks such as Prefect, Airflow, Dagster, Ray, Kubernetes, or similar
  • Strong understanding of production system fundamentals—observability, telemetry, reliability, performance, debugging, incident response, and cost management
  • Ability to reason about AI system quality beyond traditional service metrics (evaluation design, experiment reproducibility, behavioral regressions)
  • Track record of leading complex engineering initiatives and delivering impact
  • Strong technical communication skills and ability to work in a fast-paced environment
  • Fluency with modern AI and developer tooling and willingness to adopt new models, frameworks, infrastructure, and techniques

Skills

Required:PythonDistributed systemsCloud platforms (AWS preferred)Workflow/distributed execution frameworksObservability and reliabilityAI/ML experimentation infrastructureCommunication

Benefits

  • Equal Employment Opportunity employer; commitment to diversity and inclusion
  • Competitive salary determined by skills, qualifications, experience, and location
  • Opportunities for growth and taking ownership over infrastructure decisions
  • Supportive environment for learning and career development
Last updated Oct 14, 2026. Interested? Apply on the company site.

About Snorkel AI

Shaping how AI learns through better data

Snorkel AI delivers the highest quality specialized datasets for frontier LLMs and enterprise models.

AI data development servicesAboutApplied AI solutionsCareersCompanyContact usCustomer storiesData development services
Company profileWebsite

More open roles at Snorkel AI