Staff Software Engineer, Data

The Team

The Data team builds and runs the systems that make Stability AI's data usable, both to operate the business and to train our models. We're a small, collaborative team that works closely with Finance, Marketing, SRE, Security, Product and Research. We focus on outcomes rather than hours, and you'll have deep technical access and the freedom to build.

Our stack: AWS (S3, Athena, Glue, EMR, Step Functions, EventBridge), Iceberg, dbt, Postgres, Terraform, Helm, GitHub Actions, QuickSight, Stripe, Auth0, HubSpot, Customer.io, PostHog, OpenTelemetry and SLURM — or similar. We care about the judgement more than the logo.

The Work

You'll own our company data platform end to end: the AWS data lake, the transformation layer, the integrations between our business systems, and the Kubernetes-based infrastructure they run on. The data that runs the business (billing, usage telemetry, identity, CRM, ERP, cloud cost) lives in a dozen systems that were never wired together. You'll be the engineer who connects them and keeps them reliable.

You'll also be the data engineer for our research teams across audio, video, image and 3D, building the pipelines that turn raw media into training-ready datasets. This is a hands-on role with real ownership, not a data-prep or analyst seat.

Own the company data platform end to end: the S3/Athena/Glue lake, the dbt project and its daily build, ingestion from Stripe, Auth0, HubSpot, OpenTelemetry and product events, and the Postgres serving layer for dashboards and AI agents

Build and operate integrations between billing, CRM, ERP, identity and telemetry systems so revenue, usage, cost and customer identity reconcile automatically instead of through spreadsheets and tickets

Design and maintain the warehouse layer: dimensional models, data contracts with product engineering for event schemas, quality tests, lineage and a catalog

Enforce PII tiering (open vs. restricted schemas) through access grants, working with Security and Legal

Process data at scale: billions of telemetry rows, terabyte log streams, Iceberg tables, partitioning and cost control

Own orchestration and infrastructure: orchestration on EKS, Helm, Terraform, CI/CD, monitoring and alerting, and join the business-hours on-call rotation for the daily pipelines

Deliver operational reporting for Finance and leadership: revenue by business line, credits and prepaid liability, and cloud cost by department

Support the research teams across every modality: source, license-check, ingest and version large training datasets for audio, video, image and 3D

Build the generation, filtering, deduplication and captioning pipelines that turn raw or synthetic media into training-ready shards, and run them at scale on our SLURM cluster and cloud object storage

Keep lineage, quality metrics and dataset documentation current so any researcher can reproduce a training set

Lead design reviews, mentor teammates and write the runbooks that remove single-person dependencies

Partner with Finance, Marketing, SRE, Security, Product and Research on company-wide data initiatives

What You Bring

We're looking for an engineer who pairs strong software engineering habits with warehouse. The ideal candidate combines strong software engineering habits with warehouse modelling depth, and the judgement to say what a number means and what it does not.

Experience

6+ years building and running production data platforms, including ownership of the architecture

Experience integrating SaaS business systems (Stripe, HubSpot, an ERP, an identity provider) into a warehouse

Production experience preparing large multimodal training datasets and running batch workloads on HPC or cloud schedulers (SLURM, Kubernetes or similar)

Comfort working independently in a small team with real ownership

Skills & Expertise

Expert SQL and Python, with strong software engineering habits: tests, code review and CI

Warehouse and lake experience with real modelling: Iceberg or Delta on S3, Athena/Trino, Snowflake or BigQuery, and dbt

An orchestrator in production (dbt, Airflow or Argo) and distributed processing (Spark or equivalent)

AWS, Kubernetes (EKS), Helm and Terraform — you can debug a failing Glue job and a broken Helm release

Additional Strengths

Prior 0-to-1 platform ownership at a startup

Experience working with research teams on generative models, or hands-on exposure to audio, video, image or 3D data

FinOps and cloud cost reporting

Familiarity with SOC 2 or similar compliance requirements for data access

Benefits & Compensation

Health coverage for you and your family. Choose from PPO and HMO plans, including select plans where Stability AI covers 100% of premiums for you and your eligible dependents.

HSA and FSA accounts to set aside pre-tax dollars for healthcare and other eligible expenses.

Company-paid life insurance, with the option to add voluntary coverage for yourself, your spouse, and your children.

Short- and long-term disability coverage, to protect part of your income if illness or injury keeps you from working.

401(k) with company match of up to 4%.

Paid parental leave to support you as your family grows.

Flexible paid time off, plus paid company holidays.

The equipment you need to do your best work, provided by Stability AI.

At Stability AI, we aim to pay fairly and competitively, and we believe you should know what a role pays before you invest time in getting to know us. The annual base salary range for this role is $175,500 - $230,000 USD. It reflects our good-faith expectation for the role as posted and applies to candidates hired within the United States.

Where you fall within that range depends on the depth and relevance of your experience, your skills, and how you perform in our interview process. Because great candidates don’t always map neatly to a single level, we may also consider someone for a level above or below the one posted. If we do, the range will be adjusted to match, and we’ll talk it through with you openly.

Base salary is also just one part of your total compensation. Eligible employees take part in our equity program and receive the benefits described above. We’d rather talk about all of this early than leave you guessing, so your recruiter is happy to walk you through the range, how offers come together, and anything else that matters to you, at any point in the process.