Staff Software Engineer / Architect - AI

Role Description

In this opportunity as Staff Software Engineer / Architect - AI, you will:

• Lead the end-to-end architecture, design, and delivery of AI-powered solutions for law firms and corporate legal departments, from concept through production deployment.
• Review and guide system design, architecture, and code produced by AI Solution Engineers to ensure quality, scalability, security, and maintainability across engagements.
• Build sophisticated legal AI solutions, including RAG pipelines, multi-agent workflows, event-driven services, and enterprise integrations tailored to legal use cases such as contract review, litigation support, discovery, and regulatory analysis.
• Drive AI-native engineering practices across the team by embedding AI into development workflows, tooling, delivery processes, and technical standards.
• Establish observability, evaluation, and trust frameworks for production AI systems, including monitoring, logging, tracing, model evaluation, and quality benchmarking.
• Lead technical discovery with clients and legal stakeholders, translating legal workflows, business goals, and domain expertise into scalable technical architectures and solution plans.
• Mentor engineers and partner with product, engineering, and legal experts to advance reusable solution patterns, accelerate delivery, and bring emerging AI capabilities into the team.

Qualifications

• 7+ years of software engineering experience with deep expertise in system design, distributed systems, scalable architecture, and technical leadership.
• Hands-on experience building and deploying AI/ML or LLM-powered applications in production, including strong knowledge of RAG architectures, orchestration frameworks, and evaluation practices.
• Expert-level proficiency in Python and modern backend engineering, with experience designing RESTful APIs, async systems, and event-driven architectures.
• Experience with cloud platforms such as AWS, Azure, or GCP, along with Infrastructure as Code and CI/CD practices for secure, scalable deployments.
• Strong understanding of AI interoperability and agentic systems, including MCP, Agent-to-Agent workflows, third-party integrations, and tool-augmented architectures.
• Experience with observability, monitoring, and AI quality tooling such as Datadog, Grafana, Prometheus, LangSmith, Langfuse, Braintrust, or similar platforms.
• Excellent communication and stakeholder management skills, with the ability to engage credibly with engineers, legal professionals, and business leaders in ambiguous, high-impact environments.

Benefits

• Flexibility & Work-Life Balance: Flex My Way is a set of supportive workplace policies designed to help manage personal and professional responsibilities, including work from anywhere for up to 8 weeks per year.
• Career Development and Growth: Continuous learning and skill development through Grow My Way programming ensures you have the tools and knowledge to grow, lead, and thrive in an AI-enabled future.
• Industry Competitive Benefits: Comprehensive benefit plans including flexible vacation, two company-wide Mental Health Days off, access to the Headspace app, retirement savings, and tuition reimbursement.
• Culture: Award-winning reputation for inclusion and belonging, flexibility, work-life balance, and adherence to company values.
• Social Impact: Opportunities for community involvement with two paid volunteer days off annually and pro-bono consulting projects.
• Making a Real-World Impact: Helping customers pursue justice, truth, and transparency.
• Comprehensive benefits package including health, dental, vision, disability, life insurance, and a competitive 401k plan with company match.