Staff Software Engineer - AI II
Core
Define and build the AI foundation for Westlaw, including agentic systems, RAG, and large-scale knowledge services for legal professionals.
Role type
Staff Software Engineer (AI/ML)
Builds
Enterprise-scale AI systems, multi-component pipelines, autonomous AI agents, and knowledge graph architectures.
Domain
Legal technology / Generative AI / Large-scale search platforms
Deliverable
production ML models
Required skills
Python, backend service design, API design, ML principles, distributed systems, PyTorch, TensorFlow, AWS, Docker, Kubernetes, Terraform, MLOps, LLMOps
Preferred skills
Agent-based systems, vector databases, embeddings, semantic search, knowledge graphs, GraphRAG, legal domain expertise, AI governance
Technologies
PyTorch, TensorFlow, AWS, Docker, Kubernetes, Terraform, Neo4j, Amazon Neptune, OpenAI, Anthropic
Responsibilities
Design and lead enterprise-scale AI systems; drive adoption of emerging AI techniques; provide senior technical leadership and mentorship; ensure quality, security, and operational excellence; lead cross-functional collaboration to deliver business value.
