Staff AI Engineer, Enterprise Applied AI

Rivian Rivian · Auto · Palo Alto, CA +4 · Information Technology

Staff AI Engineer, Enterprise Applied AI at Rivian. This role focuses on designing, building, and operating production-grade generative AI systems to drive business value across internal teams. Responsibilities include leading technical design, defining architecture, partnering with business sponsors, and ensuring robust governance, privacy, and security. The role requires deep technical knowledge in AI/ML, hands-on experience with LLMs, RAG, embeddings, and evaluation, as well as experience in enterprise environments and compliance.

What you'd actually do

  1. Lead the technical design and hands-on development of prioritized AI applications, services, and platforms leveraging state-of-the-art LLM app stacks, retrieval-augmented generation, evaluation frameworks, and scalable serving.
  2. Define long-term architecture and engineering standards for Applied AI systems to maximize reuse, reliability, and impact across multiple product areas.
  3. Partner with business sponsors to translate high-value opportunities into roadmaps and shipped products with clear success metrics and measurable outcomes.
  4. Build a holistic view of AI investments by collaborating with adjacent engineering groups implementing AI in their domains, aligning patterns, reusing components, and avoiding duplication.
  5. Drive continuous improvement in AI methodologies and best practices; evaluate emerging capabilities and land them as secure, production-grade systems.

Skills

Required

  • BS/MS/PhD in Computer Science or a related field, or equivalent experience.
  • 8+ years in software engineering, with a proven track record delivering complex, production-ready systems in enterprise environments.
  • Deep technical knowledge in AI/ML, with hands-on experience building and deploying solutions using language models, retrieval/grounding, embeddings/vector search, and evaluation.
  • Demonstrated ability to translate ambiguous business problems into robust AI products with measurable business impact.
  • Experience defining and evolving architectures, standards, and platforms that create leverage across multiple teams.
  • Strong familiarity with security, privacy, compliance, safety, and auditability for enterprise AI systems.
  • Excellence in communication and stakeholder management; able to influence and align across diverse teams.
  • Proven curiosity and mental agility to learn and apply new technologies through hands-on development and continuous learning.

Nice to have

  • LLM app stacks
  • retrieval-augmented generation
  • evaluation frameworks
  • scalable serving
  • language models
  • retrieval/grounding
  • embeddings/vector search
  • AI literacy, enablement, and adoption
  • mentoring engineers
  • code quality, reliability, observability, and cost/performance of AI workloads
  • offline and online metrics
  • optimize latency, throughput, and cost at scale
  • make/buy decisions and vendor integrations

What the JD emphasized

  • production-grade systems
  • scalable serving
  • enterprise environments
  • measurable business impact
  • security, privacy, compliance, safety, and auditability for enterprise AI systems
  • rigorous evaluation, guardrails, and monitoring practices

Other signals

  • building and scaling production-grade systems
  • generative AI
  • language models and machine learning
  • enterprise AI solutions