Staff Software Engineer

Job Description

We're looking for a Lead Software Engineer AI (Staff Engineer AI) to join the CoCounsel team. We are building the next generation of AI agentic products for legal professionals. The team takes early Labs concepts and turns them into scalable, production-ready features.  You'll also help drive our transition to an AI-native engineering organization where AI accelerates development and engineers provide technical leadership, architecture, and judgment.

As a Lead Software Engineer AI (Staff Engineer), you will be responsible technical direction and delivery of our market leading Litigation Drafting products in CoCounsel Legal. You will be providing technical leadership for a team of talented front and back end engineers to deliver at pace.

We are looking for Leaders with a strong sense of urgency and a desire to work in a fast-paced, dynamic environment who will drive the delivery of timely solutions.  You are empowered to try new approaches and learn new technologies. You will have agency to contribute innovative ideas, create solutions, and will be accountable for end-to-end deliveries. 

Key Responsibilities:

• Build and architect AI Agentic driven systems –You will be working with the Engineering Manager to facilitate our full stack team in delivering ground breaking Agentic Legal software. 
  • Leading Technical strategy, experimentation, testing, rollout practices and operational excellence
  • Designing and building backend services using Python, FastAPI, and PostgreSQL to support real-time AI agentic interactions, with direct contribution to AI-powered capabilities such as agentic workflows, LLM orchestration, and content pipelines. 
  • Defining service boundaries, system interfaces, and scalable design patterns that form the architectural foundation of our AI platform. 
  • Supporting the front-end engineers to  deliver of  a complex Agentic UI which uses SSE and AG-UI along with Claude Agent SDK and AgentCore at run time.

• Drive innovation and technical strategy – Lead evaluation and adoption of cutting-edge AI techniques; oversee proof-of-concept work and technical feasibility for new AI-driven features; stay current with AI research and translate insights into technical roadmaps.
• Provide senior technical leadership – Lead architecture decisions and translate business requirements into scalable technical specifications; mentor senior and junior software engineers
• Ensure quality, security, and operational excellence – Design and implement testing, monitoring, and observability for AI systems at scale; lead compliance with ethical AI principles, security standards, and regulatory requirements.

• Cross-Functional Influence: Influencing and partnering closely with our Labs team, product managers, and designers to take early AI agent concepts and turn them into scalable, production-ready features for our products. Working closely with other engineering leaders to ensure consistency with other teams.

• Productionize AI Capabilities: Integrate with modern LLM providers (e.g., OpenAI, Anthropic) and implement reliable orchestration, routing, error handling, guardrails, telemetry, and deep observability for performance and reliability. Strengthen CI/CD pipelines and DevOps automation to support fast, stable, cloud‑native development workflows.

Required Qualifications

• Technical Experience: 7+ years of software engineering experience, including 2+ years in technical leadership roles and 3+ years building scalable cloud applications, APIs, and AI-enabled products. Proven track record of shipping production-grade AI systems to customers, including deployment, monitoring, evaluation, scaling, and ongoing operational ownership in enterprise environments.
• Technical Skills: Strong proficiency in Python, FastAPI, PostgreSQL, and AWS or Azure cloud services, with a deep understanding of distributed systems, architecture, and data modelling. building and owning CI/CD pipelines for production web applications (GitHub Actions, GitLab CI, CodePipeline, or similar), including automated testing and safe release practices

• Collaboration & Communication: Comfortable operating in fast-paced, iterative environments, with strong cross-functional communication skills. Strong product sense and the ability to work from ambiguous requirements.
• Leadership: Strong leadership and mentoring experience; excellent strategic thinking and problem-solving; outstanding communication and presentation skills with experience influencing senior stakeholders.

Preferred Qualifications 

• BA or BS in Computer Science or equivalent practical experience 
• Strong knowledge of vector databases, embeddings, semantic search, and information retrieval;
• Hands-on experience with the AG-UI protocol or comparable agent-UI protocols and SDKs. Strong proficiency in SSE events
• Knowledge of backend streaming patterns (Node, Python/FastAPI, etc.) sufficient to debug end to end
• Experience with Datadog (logs/RUM/traces), CloudWatch Log Insights, and Braintrust

• Prior experience working on products for legal, compliance, or other regulated professional domains 

#LI-SB1

What’s in it For You?

• Hybrid Work Model: We’ve adopted a flexible hybrid working environment for our office-based roles while delivering a seamless experience that is digitally and physically connected.
• Flexibility & Work-Life Balance: Flex My Way is a set of supportive workplace policies designed to help manage personal and professional responsibilities, whether caring for family, giving back to the community, or finding time to refresh and reset. This builds upon our flexible work arrangements, including work from anywhere for up to 8 weeks per year, empowering employees to achieve a better work-life balance.
• Career Development and Growth: By fostering a culture of continuous learning and skill development, we prepare our talent to tackle tomorrow’s challenges and deliver real-world solutions. Our Grow My Way programming and skills-first approach ensures you have the tools and knowledge to grow, lead, and thrive in an AI-enabled future.
• Industry Competitive Benefits: We offer comprehensive benefit plans to include flexible vacation, two company-wide Mental Health Days off, access to the Headspace app, retirement savings, tuition reimbursement, employee incentive programs, and resources for mental, physical, and financial wellbeing.
• Culture: Globally recognized, award-winning reputation for inclusion and belonging, flexibility, work-life balance, and more. We live by our values: Obsess over our Customers, Compete to Win, Challenge (Y)our Thinking, Act Fast / Learn Fast, and Stronger Together.
• Social Impact: Make an impact in your community with our Social Impact Institute. We offer employees two paid volunteer days off annually and opportunities to get involved with pro-bono consulting projects and Environmental, Social, and Governance (ESG) initiatives.
• Making a Real-World Impact: We are one of the few companies globally that helps its customers pursue justice, truth, and transparency. Together, with the professionals and institutions we serve, we help uphold the rule of law, turn the wheels of commerce, catch bad actors, report the facts, and provide trusted, unbiased information to people all over the world.