Software Engineer — AI Infrastructure

About SnorkelAt 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 since 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!The AI Infrastructure team within Snorkel owns the platform layer that powers everything at Snorkel — and that layer is evolving. Beyond the data platform (pipelines, access layers, event systems, governance, compute), we are now building Snorkel's AI infrastructure: the foundational agentic stack that will let every team at Snorkel build, run, and govern AI agents, and the LLM efficiency layer that keeps AI costs under control as usage scales.We are a small team with a large surface area, in the middle of two foundational shifts: moving from a single-database data path to a multi-source, event-driven platform (Postgres/RDS, Snowflake, S3, metrics platform), and moving from bespoke, one-off agent implementations to a shared, governed, agentic-first platform. The decisions being made now will define how data and agents operate at Snorkel for years. You will be making them.You'll also shape our AI-native development workflow, contribute to modernizing CI/CD (Buildkite, GitHub Actions), and integrate AI SRE tooling. Your work will directly accelerate developer velocity, reliability, and product quality across the company.What You'll DoBuild the Agentic Factory Foundation. Design and build the opinionated agentic stack that FDEs, delivery, and product engineering teams will use to scaffold agent workflows: a common orchestration layer for defining and running agents, a memory layer (short-term working memory and long-term persistence), a context graph / knowledge layer grounding agents in project, spec, and platform state, an MCP gateway providing secured, governed, auditable tool access, an evaluation layer testing agents against trace-level and outcome-level criteria, and an observability layer for traces, feedback, metrics, and cost. Start pragmatic — leverage existing building blocks to ship real use cases (self-healing agents, spec-to-eval pipelines, debugging agents) before going deep on every layer.Build the LLM cost and efficiency platform. LLM token spend is growing with the business, and controlling it is a first-class engineering problem. Build the queuing and throttling layer that governs synchronous LLM requests, async and batched call paths for workloads that don't need real-time responses, token optimization (prompt compression, caching, model routing), token usage metering and attribution so teams can see what they spend and why, and world-model approaches that let agents reuse knowledge instead of re-querying models.Build the foundational data access layer and SDKs. Design the shared access library that Platform, Packaging, and Dataset API teams use to read from and write to multiple data sources (Snowflake, S3, RDS) — abstracting entity data from the specific infrastructure underneath so we can scale and improve infrastructure without every product team absorbing the change. Interfaces provide built-in auth, RBAC enforcement, pagination, and query governance.Design and implement event-driven data flows using event brokers, CDC connectors, schema registry, event routing, and dead letter queues. Make sure events flow reliably and failures are visible and recoverable.Build governance, lineage, and audit infrastructure — for data and for agents. Track how data moves through the platform, enforce who (and which agent) can access what, and log what happened. This includes PII handling, retention policy enforcement, and audit infrastructure for enterprise and federal compliance, extended to agent actions and tool calls through the MCP gateway.Own reliability and cost. Instrument the platform with OpenTelemetry, define and monitor SLOs for query latency, pipeline success rates, and agent workflow health, and build alerting that catches issues before they become incidents. Contribute to cost visibility and optimization across both infrastructure (query cost estimation, workload right-sizing, storage tiering) and AI spend (token cost attribution, model routing). You will be on-call for the systems you build.What You'll Bring2+ years building platform infrastructure, data infrastructure, or backend systems with significant data components. You have built and operated pipelines, data access layers, or production services other teams depend on.Strong proficiency in Python. Our stack is Python-heavy across Prefect, FastAPI, dbt, and the SDK layer.Hands-on experience building with LLMs in production — working with LLM APIs, and reasoning about tokens, context windows, rate limits, batching, and caching. You understand why an async batched call costs less than a sync one and can design systems around that.Fluency with AI-assisted development tools (Claude Code, Cursor, or similar). This is a hard requirement — the team uses these tools daily and we expect engineers to leverage them for code generation, debugging, and investigation.Hands-on experience with SQL and at least two of: Snowflake, Redshift, Postgres. You understand the performance characteristics of each and can write queries that don't bring down production.Experience with AWS — S3, RDS, EKS, EventBridge, IAM. Comfortable working in a Terraform-managed environment.Experience with Kubernetes. Our workloads run on EKS and you will deploy, debug, and scale services on K8s.Familiarity with data orchestration tools (Prefect, Airflow, or Dagster) and transformation frameworks (dbt).Understanding of data governance concepts — RBAC, PII handling, audit logging, data lineage — and interest in extending them to agent and tool-call governance.Nice to HaveExperience building agentic systems or infrastructure — agent orchestration frameworks, MCP servers/gateways, agent memory or knowledge layers, or agent evaluation harnesses.Experience with LLM serving/gateway infrastructure — request queuing, rate limiting, model routing, semantic caching, or token cost optimization at scale.Experience building shared libraries or SDKs consumed by multiple teams — versioning, backwards compatibility, migration support.Experience with event-driven architectures — CDC, event buses, schema registries, at-least-once delivery semantics.Experience with OpenTelemetry, ClickHouse, or similar observability infrastructure, including LLM/agent trace observability.Prior work in regulated environments (SOC 2, FedRAMP, HIPAA) where compliance requirements shaped system design.Experience with Ray for distributed compute workloads.Actual compensation will be determined based on factors including skills, qualifications, experience, and geographic location.Salary range(s) for this role$180,000—$240,000 USDBe Your Best at SnorkelJoining 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.