Profile cover
A

AI Feature Query Cost Optimizer

Aubrey Okafor

Full-time · Job · Los_Angeles

About the role

The feature queries you'll optimize run millions of times a day across every model our platform serves, and even small inefficiencies compound into real infrastructure cost at that scale. We need a senior engineer to profile our highest-cost feature queries and rework them for both speed and warehouse spend, working across Spark and Snowflake without breaking the freshness guarantees downstream models depend on.

Responsibilities

  • Profile and optimize the highest-cost feature queries in production
  • Rework Spark and Snowflake queries for both speed and cost
  • Maintain feature freshness guarantees while optimizing execution
  • Report warehouse cost savings to platform and finance stakeholders

Requirements

  • 6+ years working with Spark and cloud data warehouses at scale
  • Deep understanding of query optimization and cost tradeoffs
  • Experience with feature stores or online/offline feature parity
  • Comfortable owning cost outcomes with measurable targets

Benefits

  • Fully remote position
  • Top-of-market salary + equity
  • Health and dental insurance
  • Annual GPU/data conference budget

Keep exploring related roles

Use these topic pages to discover more roles in the same hiring cluster.

Job Type

Full-time

Level

Senior

Language

English

Salary Range

$200k+ / year

AI Expertise

MLOps & AI Infrastructure

Skills

python spark databricks snowflake sql

Ready to apply for this role?

Create a free talent account in under 2 minutes.

  • Apply to verified AI companies
  • Get AI-matched job recommendations
  • Message hiring managers directly
  • Build your public AI talent profile

Interested in this role? Create your free profile to apply.

Apply — create free profile