Lead AI Engineer

Location
Remote (US, UK, or Brazil), ET timezone
Experience Level
Lead
Type Of Employment
Full-time
Remote
Yes
Domain
LegalTech

We are seeking a Lead AI Engineer to drive the development of high-impact AI/ML systems, including Retrieval-Augmented Generation (RAG), knowledge graphs, LLM integrations, and agentic AI frameworks. This is a leadership role at the forefront of applied AI, shaping the core of a fast-scaling SaaS product in a document- and workflow-heavy legal environment.

Our client is an innovative legal technology company developing cutting-edge, AI-powered solutions for law firms and legal departments. Operating across the US, Brazil, and UK, the Client is on a mission to optimize legal workflows and decision-making with intelligent, scalable tools.


Responsibilities

  • Design & Deploy Scalable AI Systems: Lead the development of RAG systems, knowledge graph pipelines, and agent-based AI solutions. Translate research into production-grade features delivering real-world value.
  • Team Leadership & Enablement: Establish best practices, workflows, and tooling to support a high-performing AI/ML team. Foster close collaboration with engineering, product, and legal subject-matter experts.
  • Cross-Functional Collaboration: Partner with product managers and business stakeholders to ensure alignment, impact-driven prioritization, and delivery of AI features that enhance user outcomes.
  • Agile Execution in a Startup Environment: Operate effectively with limited resources, adapt quickly to pivots, and drive progress through a fast-paced, iterative development cycle.
  • Innovation & Technical Excellence: Stay at the leading edge of AI and machine learning developments. Proactively experiment with and implement emerging technologies and approaches.

Requirements

  • Proven experience in building and deploying RAG systems, knowledge graphs, and LLM-based features at scale
  • Deep understanding of agentic AI architectures and real-world implementation
  • Track record of leading AI/ML initiatives within high-growth SaaS or AI-native startups (B2B preferred)
  • Strong organizational skills with a hands-on approach to technical leadership
  • Fluent or high-speaking proficiency in English
  • High emotional intelligence, integrity, and collaborative mindset
  • Ability to work across teams and disciplines to deliver user-centric outcomes

Nice to Have

  • Experience in legal, compliance, or document-centric SaaS products
  • Background in contract analytics, document automation, or workflow orchestration
  • Familiarity with scaling AI in regulated or domain-specific environments

Tech Stack

  • AI/ML: Python, LlamaIndex, Qdrant, CrewAI
  • Backend: Python, Django, AWS RDS
  • Frontend: TypeScript, Next.js
  • Infrastructure: AWS, Terraform
Date posted: July 7, 2025