Data Engineer

About Baseten

Baseten powers mission-critical inference for the world's most dynamic AI companies, like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma and Writer. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production. We're growing quickly and recently raised our $1.5B Series F, led by Altimeter Capital, Conviction Partners, and Spark Capital. Join us and help build the platform engineers turn to to ship AI products.

About the role

We're hiring a Data Engineer to build and scale Baseten's internal data platform. This role sits at the intersection of data engineering, analytics, and data science, transforming raw product and business data into reliable datasets that power decision-making. You'll design the data models, pipelines, and analytics infrastructure that enable teams across Product, Engineering, Finance, Marketing, and Sales to understand usage and performance. This includes working with AI inference, infrastructure, and observability data to generate insights about the product, business operations and platform economics. You'll partner closely with stakeholders to build robust, scalable pipelines, define company-wide metrics that inform strategy and planning.

Responsibilities

• Design and maintain core data models and semantic layers
• Develop and orchestrate batch and streaming data pipelines using technologies such as Apache Beam, Kafka, Airflow, or similar frameworks
• Analyze inference and infrastructure telemetry, including data from OpenTelemetry, Grafana, and other observability tools
• Define and maintain company-wide metrics across product usage, performance, and customer lifecycle
• Enable self-service analytics through agents and tools, with well-structured semantic layers and context
• Ensure data reliability and quality through testing, documentation, and governance

Preferred qualifications

• Understanding of inference metrics such as latency, throughput, token usage, and model performance
• Experience supporting B2B SaaS and/or consumption-based platforms
• Application of forecasting and predictive modeling (e.g., ARIMA, Prophet) to business processes

Benefits

• Competitive compensation, including meaningful equity
• 100% coverage of medical, dental, and vision insurance for employee and dependents
• Flexible PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)
• Paid parental leave
• Fertility and family-building stipend through Carrot
• Company-facilitated 401(k)
• Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities