← All jobs · Baseten

Data Engineer

Baseten ·
53
AI-Agency
B55 U50
📍 San Francisco, US Mid
Apache BeamKafkaAirflowOpenTelemetryGrafana
TL;DR

Data Engineer at Baseten building internal data platforms and analytics infrastructure for an AI inference company. Design data models, pipelines, and metrics to support product, engineering, and business teams.

Apply at Baseten →
share:
you'll be redirected to the company's career page

Job description

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 $300M Series E, backed by investors including BOND, IVP, Spark Capital, Greylock, and Conviction. Join us and help build the platform engineers turn to to ship AI products.

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

PREFERRED QUALIFICATIONS

BENEFITS

Apply now to embark on a rewarding journey in shaping the future of AI! If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you.

At Baseten, we are committed to fostering a diverse and inclusive workplace. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, or veteran status.

We are an Equal Opportunity Employer and will consider qualified applicants with criminal histories in a manner consistent with applicable law (by example, the requirements of the San Francisco Fair Chance Ordinance, where applicable).

Apply at Baseten →

More open roles at Baseten

Baseten · 🔄 synced 3h ago
Post-Training Applied Researcher
📍 San Francisco, US 🛠 AI tools welcome at work · Senior
Post-training applied researcher at Baseten building reward functions, training pipelines, and eval harnesses to fine-tune open-source LLMs for production use. Work directly with customers on domain-specific model training spanning healthcare, code generation, and agentic workflows.
PyTorchJAXGRPODPOPPORL
89
AI-core
Baseten · 🔄 synced 3h ago
Post-Training Research Scientist
📍 San Francisco, US · Senior
Post-training research scientist at Baseten, an AI inference platform. Role spans pure research on model learning and alignment, plus applied work on training methodology and inference optimization for production systems serving companies like Cursor and Notion.
PyTorchJAXTensorFlowvLLMTriton
85
AI-core
Baseten · 🔄 synced 3h ago
Software Engineer - AI Enablement
📍 San Francisco, US 🛠 AI tools welcome at work · Mid
Software engineer at Baseten building internal AI agents and LLM-powered workflows to boost engineering productivity. Focus on evaluating, deploying, and customizing AI coding assistants and custom agents for the engineering org.
PythonLLM agentsClaude CodeCursorCodex
80
AI-core
Baseten · 🔄 synced 3h ago
Forward Deployed Engineer
📍 San Francisco, US 🛠 AI tools welcome at work · Mid
Forward Deployed Engineer at Baseten, an AI inference platform. You'll architect and deploy production AI applications for customers like Cursor and Notion, combining hands-on coding with customer-facing implementation and product work.
PythonDockerML model deploymentAPI development
77
AI-core
Baseten · 🔄 synced 3h ago
Software Engineer - Training Product
📍 San Francisco, US 🛠 AI tools welcome at work · Senior
Software engineer at Baseten building training infrastructure for AI companies. Own features like multi-node training and serverless RL from conception to production, working across API, backend, and infrastructure layers.
PyTorchKubernetesvLLMNCCLMegatronDeepSpeed
76
AI-core