Data Engineer, Imbue

$170-350k

$10,000–$2,000,000 in equity

GCP

Airflow

Redshift

Spark

Amplitude

Snowflake

Flink

Senior and Expert level

San Francisco Bay Area

Training AI in reasoning

Be an early applicant

Training AI in reasoning

21-100 employees

B2BArtificial IntelligenceDeep TechMachine LearningSaaS

Be an early applicant

$170-350k

$10,000–$2,000,000 in equity

GCP

Airflow

Redshift

Spark

Amplitude

Snowflake

Flink

Senior and Expert level

San Francisco Bay Area

21-100 employees

B2BArtificial IntelligenceDeep TechMachine LearningSaaS

Company mission

To empower humans in an age of AI by creating powerful computing tools for people.

Job

Company

Role

Who you are

If you’re excited about turning messy product data into actionable insights, and building systems that can scale with our research, we’d love to get connected!
A strong software engineer with 5+ years of experience, ideally working with large-scale data systems
Experienced in designing and maintaining data pipelines and infrastructure, especially for analytics, experimentation, and ML
Comfortable with tools for data orchestration (Airflow, Prefect), batch or streaming processing (Spark, Ray, Flink), and event tracking and analytics (Amplitude, PostHog)
Experienced with cloud-based infrastructure and storage (e.g., S3, GCP, Snowflake, or Redshift), and thoughtful about cost-performance tradeoffs
Exposure to MLOps, model serving infrastructure, or ML workflows
Pragmatic and principled! You know when to optimize and when to ship

What the job involves

We're looking for a Data Engineer to build and own the pipelines and data infrastructure that power our product and research efforts. Your work will directly support model training, evaluation, product analytics, and safety systems. You’ll partner closely with team members building our coding agents to make sure we’re capturing the right signals and using them well
Combine synthetic data generation with human annotation platforms to produce high quality data that advances our product and research roadmap
Design and build resilient, scalable pipelines (ETL and ELT) for batch and streaming data
Develop and maintain infrastructure to support self-serve analytics, experimentation, and dataset generation. Prototype, evaluate, and make “build vs buy” decisions
Help define and improve data modeling practices across the company, including instrumentation standards, dimensional modeling for analytics and feature stores for machine learning (ML)
Build integrations with ML infrastructure to support training pipelines, inference logging, and model monitoring (MLOps)
Debug pipeline failures, automate deployment processes, and improve data quality and reusability

Application process

All submissions are reviewed by a person, so we encourage you to include notes on why you're interested in working with us. If you have any other work that you can showcase (open source code, side projects, etc.), certainly include it!

Our take

Since the widened attention placed on generative AIs such as Chat GPT, it seems that investors have intensified their interest in general intelligence. Imbue is one of the most recent AI research and training companies to step into the limelight after it achieved unicorn status in a $200 million Series B funding round in 2023.

With competition including the likes of OpenAI and Anthropic, Imbue researches and improves the reasoning of AI models. Its focus on foundational AI means that its training has a wide range of lucrative applications. Using a full-stack approach (essentially building its tech from scratch) enables the creation of feedback loops that gradually develop even better AI models.

Freddie

Company Specialist at Welcome to the Jungle

Company

Company HQ

SoMa, San Francisco, CA

Leadership

Kanjun Qiu

(Co-Founder & CEO)

Current General Partner at Outset Capital and Sequoia Capital Scout. Previously CEO and Co-Founder at Sourceress and Member at South Park Commons.

Josh Albrecht

(CTO)

Current General Partner at Outset Capital. Previous CTO at Sourceress, Member at South Park Commons, and Public Data Lead at Addepar.

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 4 more jobs at Imbue