Staff Python Engineer

The Role

We’re looking for a Staff Python Engineer to join our AI Engineering & Infrastructure team to bring our latest AI-driven features and services into production. Deploying them at scale, improving infrastructure, and ensuring robustness in production. You’ll work closely with AI researchers, software engineers, and product teams to bridge the gap between cutting-edge AI research and real-world applications.

Responsibilities

• Developing production-ready APIs and services that expose AI functionality to internal and external applications.
• Improving reliability & monitoring for AI-driven applications in production.
• Scaling AI systems to handle real-world legal use cases (e.g., legal document analysis, case predictions, automated legal advice).
• Collaborating with AI engineers to ensure smooth deployment of AI workflows and models into production.
• Working with event-driven architectures and async workflows to process large-scale AI workloads efficiently.
• Ensuring security & compliance in AI-driven legal services.

Requirements

• Strong Python experience in building scalable backend systems.
• Familiarity with API design & distributed systems architecture.
• Experience working with event-driven architectures (e.g. Kafka, Pub/Sub, AWS Step Functions, etc.).
• Experience working with cloud platforms (AWS, GCP etc).
• Understanding of best practices in observability, monitoring, and debugging.

Nice to Have

• LLM Observability & Evaluation – familiarity with tools such as Langfuse for monitoring model generations, managing prompts, and measuring quality in production.
• Comfortable optimising performance & scaling distributed AI workloads and ML Ops experience.
• Full-stack Typescript experience.
• Hands-on work with vector databases, hybrid retrieval methods, and evaluation of retrieval quality.
• Agentic & Orchestrated Systems – exposure to multi-step reasoning, agent frameworks, or orchestration tools (e.g. LangChain, AutoGen, Inngest, Temporal) where LLMs call tools, plan tasks, or coordinate workflows.
• Experience working in collaboration with researchers so that new models, pipelines, or research outputs can be integrated and evolved iteratively.
• Prior Experience in Legal Tech - understanding of the legal industry and experience working with legal technology solutions.

Interview process

• Introductory call with our Talent team
• 1:1 with your hiring manager
• Live technical assessment - systems design and pairing with two of our engineers
• Values interview with our CTO/Co-founder
• We offer!