As a Staff/Lead Data Scientist focused on Context Engineering at Rivian, you will play a crucial role in building the semantic and data infrastructure for AI agents within the Customer organization. This position involves designing the knowledge layer that AI agents will utilize, ensuring a coherent and AI-consumable ecosystem across Marketing, Sales, and Fulfillment data. You will collaborate closely with various teams to architect data models and frameworks that enhance the functionality of AI agents. This foundational role requires a blend of data architecture, knowledge engineering, and applied AI expertise.

• Bachelor's degree in Computer Science, Information Systems, Data Engineering, or a related technical field; Master's degree preferred.
• 8+ years of experience in Data Architecture, Systems Engineering, or Data Engineering.
• Proven experience building semantic layers, knowledge graphs, or data ontologies.
• Expert-level command of vector databases and orchestration frameworks.
• Deep mastery of the Databricks ecosystem including Unity Catalog, Delta Lake, and Vector Search.

• Design, develop, and own the semantic layer and business ontology for the Customer domain.
• Architect a unified 'Customer 360' schema that harmonizes disparate data streams.
• Design the Retrieval-Augmented Generation (RAG) infrastructure for indexing and retrieving unstructured data.
• Build the logic for persistent identity resolution and conversation state for AI agents.
• Architect the Action Layer for secure APIs and function calls across various applications.
• Establish protocols for data privacy, security, and latency for AI agents.

• Employees at Rivian are often offered comprehensive health, dental, and vision insurance, a 401(k) with company match, ESPP and potential RSU grants, performance bonuses, and generous time off including paid parental and sick leave. Perks can include 24/7 mental-health coaching and therapy, Hinge Health for Anthem enrollees, tuition assistance and professional development, plus on-site food and a wide range of employee discounts on items like computers, mobile phones, home loans, and pet care.