Data Scientist

Rivian Staff/Lead Data Scientist Focused On Context EngineeringRivian is on a mission to keep the world adventurous forever. This goes for the emissions-free electric adventure vehicles we build, and the curious, courageous souls we seek to attract. As a company, we constantly challenge what's possible, never simply accepting what has always been done. We reframe old problems, seek new solutions and operate comfortably in areas that are unknown. Our backgrounds are diverse, but our team shares a love of the outdoors and a desire to protect it for future generations.Rivian's AI team sits within the analytics function of the customer organization—the team that spans marketing, sales, and market intelligence—and is building the semantic and data infrastructure that makes intelligent agents possible at scale. As a staff/lead data scientist focused on context engineering, you will design the knowledge layer that AI agents reason from—owning the ontologies, semantic models, identity resolution logic, and retrieval architecture that unify Rivian's marketing, sales, and fulfillment data into a coherent, AI-consumable ecosystem. This is a foundational systems role: the agents are only as good as the context you build. You'll work at the intersection of data architecture, knowledge engineering, and applied AI, partnering closely with the agentic solutions team, data platform, martech, and the broader customer org analytics team.ResponsibilitiesCustomer Domain Ontology: Design, develop, and own the semantic layer and business ontology—mapping definitions, relationships, and business logic across marketing, sales, and fulfillment. Ensure AI agents have a standardized, enterprise-wide conceptual framework to reason accurately and consistently.Cross-Functional Data Modeling: Architect a unified 'customer 360' schema that harmonizes disparate data streams from marketing, sales, and fulfillment into a single, AI-consumable source of truth.RAG & Vector Architecture: Design the retrieval-augmented generation (RAG) infrastructure, determining how unstructured data (e.g., shipping logs, sales notes, brand guidelines) is indexed, chunked, and retrieved for agentic context.Identity & State Orchestration: Build the logic for persistent identity resolution and conversation state, ensuring an AI agent can maintain customer context as they move from a sales inquiry to a fulfillment update.Agentic Tooling & API Design: Architect the action layer—the secure framework of APIs and function calls that allow agents to execute tasks across Salesforce, Braze, and specialized in-house applications.AI Data Governance: Establish protocols for data privacy, security, and latency, ensuring agents only access authorized data and respond with sub-second retrieval times.QualificationsEducation: Bachelor's degree in computer science, information systems, data engineering, or a related technical field; master's degree preferred.Experience: 8+ years in data architecture, systems engineering, or data engineering, with a proven track record of designing complex, multi-source data ecosystems.Semantic Modeling & Ontologies: Proven experience building semantic layers, knowledge graphs, or data ontologies. Ability to translate abstract business rules and cross-departmental relationships into structured data definitions that LLMs can naturally interpret.Architectural Vision: A systems-thinking mindset; you can map how a change in a fulfillment status ripples through the data layer to inform a marketing re-engagement agent.Databricks Expertise: Deep mastery of the Databricks ecosystem (Unity Catalog, Delta Lake, Vector Search) to manage the end-to-end data lifecycle for AI.Technical Mastery: Expert-level command of vector databases and orchestration frameworks (e.g., LangChain, LlamaIndex). Expert-level Python and strong SQL. You understand the plumbing required to connect LLMs to enterprise data.MarTech & CRM Literacy: Deep functional knowledge of how data is structured within CDP platforms, CRM systems (e.g., Salesforce), and behavioral event streams to support cross-departmental agent hand-offs.Internal Systems Integration: Experience architecting data flows between modern cloud stacks and proprietary in-house applications, ensuring seamless bi-directional communication for AI agents.Environment: Comfortable operating in a fast-paced, high-ambiguity environment with strong attention to detail, a builder's mentality, and the ability to make principled architectural decisions with incomplete information.Ability to stand, sit, or walk for 8-10 hours per day. Required to communicate using phone and/or e-mail. Ability to view, read, and interpret documents. Ability to perform all duties in an office environment that may contain ambient noise and temperature fluctuations.Pay disclosure: The salary range for this role is $154,000 to $192,500 for Irvine, CA and El Segundo, CA based applicants. This is the lowest to highest salary we in good faith believe we would pay for this role at the time of this posting. An employee's position within the salary range will be based on several factors including, but not limited to, specific competencies, relevant education, qualifications, certifications, experience, skills, geographic location, shift, and organizational needs. The successful candidate may be eligible for annual performance bonus and equity awards. We offer a comprehensive package of benefits for full-time and part-time employees, their spouse or domestic partner, and children up to age 26, including but not limited to paid vacation, paid sick leave, and a competitive portfolio of insurance benefits including life, medical, dental, vision, short-term disability insurance, and long-term disability insurance to eligible employees. You may also have the opportunity to participate in Rivian's 401(k) Plan and Employee Stock Purchase Plan if you meet certain eligibility requirements. Full-time employee coverage is effective on their first day of employment. Part-time employee coverage is effective the first of the month following 90 days of employment. More information about benefits is available at rivianbenefits.com.