AI tools, scored for your job
Get paid to train AI
Hiring now · posted today Tech Full-time · Remote

Staff Software Engineer, Data

at Stability AI

Pay not published

The rate is shared during the application.

Remote · Remote Language: English Posted 27 Oct 2026

About this role

Preparing for this role? See the best AI tools for developers, practise with ready-made prompts for developers, read Will AI Replace Developers? What 2026 Data Says and take the free 30-day AI challenge.

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.

This is a remote role. Location as listed by Stability AI: Remote.

How applying works

  1. Click Apply now and enter your email.
  2. You go straight to the application. We email you a copy of this role and your link.
  3. You apply on Stability AI's own careers page. Zekai is not the employer and never charges you.
How to get this job

Your plan for “Staff Software Engineer, Data”

Are you ready? Check what you have

Taken from this listing's own requirements.

0 / 4

What to know before you apply

Credentials
The listing asks for an active professional licence, 6+ years of experience. Put it in the first line of your profile: reviewers screen for it before anything else.
Timing
This listing is new today. Projects can close once they have enough people, so it is worth applying while it is open.
How to stand out for developers
  • Mirror the job's own wording in your CV: reviewers and screening tools look for the skills listed above.
  • Show real work: a portfolio, GitHub, case study or numbers from past projects beats a long list of duties.
  • Show how you already use AI tools in your work. AI companies hire people who use their products well.
  • Apply early and follow up once, politely, if you hear nothing in two weeks.
Prepare free with Zekai

3 steps to stand out as one of the best developers

  1. 01
  2. 02

    Practise with real prompts

    Many tests ask you to write or judge prompts in your field. Practise on these first.

    The Project Scaffolding Agent

    / ├── app/ │ ├── layout.tsx │ └── page.tsx ├── components/ │ └── ui/ ├── lib/ │ └── utils.ts ├── public/ ├── package.json ├── tsconfig.json ├── next.config.mjs └── tailwind.config.ts

    Boilerplate scaffolding and unit test generation

    Act as a senior software engineer. Generate a modular [[LANGUAGE]] class or function for [[FEATURE_DESCRIPTION]], following clean code principles. Include error handling for [[EDGE_CASES]] and write comprehensive unit tests using…

    Prompt library for developers →
  3. 03

    Take the free 30-day AI challenge

    Five minutes a day to get confident with AI in your own work. Day 1 is open now, no card needed.

    Start Day 1 free →

    Guides for developers

Developers

More paid AI roles for developers

Confidential clientAI training · Developers

Software Engineering Specialist

Top pay
  • Remote · United States, United Kingdom +1
  • Contract · flexible hours

Our partner is engaging Software Engineering Specialists to contribute technical expertise to a customer project focu…

$150–$350/hr
33d ago
Confidential clientAI training · Developers

Senior Software Engineering Experts

NewTop pay
  • Remote · United States
  • Contract · flexible hours

Our partner is working with a leading AI research lab on a project to improve how AI systems do real-world software e…

$120–$180/hr
New today
Confidential clientAI training · Developers

Open Source Contributor (GitHub)

Top pay100 open
  • Remote · Worldwide
  • Contract · flexible hours

We are looking for highly skilled Open-Source Software Engineers to contribute to an AI training project involving so…

$100–$150/hr
41d ago
Confidential clientAI training · Developers

Software Engineer — Agentic Search Systems

Top pay
  • Remote · United States
  • Contract · flexible hours

We're looking for engineers who have shipped production search systems (especially agentic ones now that we're in the…

$80–$150/hr
88d ago
Confidential clientAI training · Developers

Software Engineering Expert

Top pay
  • Remote · United States
  • Contract · flexible hours

The research teams behind the best-known AI models come to our partner for software engineering judgment they can't g…

$70–$150/hr
43d ago
Confidential clientAI training · Developers

Frontend Engineer Talent Network

Top pay
  • Remote · United States
  • Contract · flexible hours

Join our Frontend Engineer Expert Network to connect with leading AI labs and companies seeking your expertise. This …

$70–$150/hr
249d ago
new in the last 24 hours

New roles in your inbox every morning

Pick your field. One short email a day, only when new roles open. Unsubscribe any time.

Roles come from public listings and change daily. Pay is what the listing shows; offers depend on your test, experience and country. If you join through a Zekai link, the platform may pay Zekai a fee. Your pay stays the same.

Staff Software Engineer, Data
Get paid to train AI →

See Zekai first in Google