Staff Software Engineer – AI (Westlaw AI)
Core
Spearhead development of cutting-edge AI-powered solutions and guide technical strategy for enterprise-grade AI systems transforming customer value delivery.
Role type
Staff Software Engineer – AI (LLM/Agent Systems)
Builds
Enterprise-scale AI systems, multi-component pipelines, RAG architectures, autonomous AI agents, and production software integrations.
Domain
Legal, tax, fraud, and compliance technology within the professional services industry.
Deliverable
production ML models | product features | infrastructure
Required skills
Python, PyTorch, TensorFlow, JAX, LLM architectures, distributed agent system design, vector databases, semantic search, MLOps, LLMOps, API design, microservices, system scalability, AI governance, model risk management.
Preferred skills
Master's degree in CS/Engineering, domain expertise in legal/tax/accounting, track record of AI product launches, publications/patents in AI/ML.
Technologies
PyTorch, TensorFlow, JAX, OpenAI, Anthropic, vector databases, semantic search frameworks.
Responsibilities
Design and lead implementation of complex enterprise-scale AI systems; Architect scalable AI model integration strategies; Lead adaptation and customization of AI models for specialized domains; Define technical solutions for AI system architecture; Lead evaluation and adoption of cutting-edge AI technologies; Oversee proof-of-concept projects; Mentor senior and junior Software Engineers; Champion new MLOps and LLMOps practices; Design and implement testing frameworks and observability solutions; Lead compliance initiatives for ethical AI and regulatory requirements.
Seniority
Staff, hands-on IC with strategic leadership and mentorship responsibilities.