Staff Software Engineer – AI (Westlaw AI)

Thomson Reuters • Ann Arbor, MI

74d • $147,000 - $273,000

About The Position

As a Staff Software Engineer – AI, you will spearhead the development of cutting-edge, AI-powered solutions and drive the design of enterprise-scale AI systems that transform how our organization delivers value to customers. You will build and architect AI-driven systems, design and lead the implementation of complex, enterprise-scale AI systems, including sophisticated multi-component architectures and ML capabilities. You will also architect scalable AI model integration strategies for production software through robust APIs, distributed systems, and cloud platforms. Additionally, you will define technical standards and best practices for AI systems within Westlaw AI and across Thomson Reuters.

Requirements

• Bachelor’s degree in computer science, Engineering, or equivalent experience; master’s degree preferred.
• 8+ years of experience building AI/ML systems, with extensive hands-on experience in frameworks such as PyTorch, TensorFlow, or JAX.
• Deep understanding of machine learning principles and lifecycle management for enterprise-scale AI applications.
• Demonstrated expertise integrating AI models into large-scale production systems using AWS, Azure, or GCP, leveraging tools like Docker, Kubernetes, and Terraform.
• Strong leadership experience driving technical architecture, mentoring teams, and influencing strategic engineering direction.
• Outstanding communication and presentation skills with experience engaging senior stakeholders.

Nice To Haves

• Experience leading teams in building AI-driven systems, agent-based architectures, or enterprise integrations with AI.
• Domain expertise in legal, tax, accounting, or other regulated industries.
• Proven experience with AI governance, model risk management, and responsible AI principles.
• Track record of end-to-end delivery of AI-based solutions and AI product development.
• Prior exposure to Westlaw or similar legal research and information retrieval systems is highly desirable.

Responsibilities

• Lead the evaluation and strategic adoption of cutting-edge AI techniques to solve complex business challenges.
• Oversee and assess emerging technologies, translating insights into actionable technical roadmaps.
• Drive innovation initiatives and establish Westlaw AI as a leader in applied AI for legal, tax, and compliance technology.
• Lead technical architecture decisions and translate high-level business requirements into comprehensive, scalable technical solutions.
• Champion MLOps and LLMOps practices across the organization, leading both design and implementation initiatives.
• Represent engineering in cross-functional discussions, balancing innovation with operational excellence.
• Design and implement comprehensive testing frameworks, monitoring systems, and observability solutions for AI models and services.
• Conduct advanced systems analysis and drive operational improvements across AI infrastructure.
• Establish and enforce best practices in reliability, scalability, and performance tuning.
• Partner with AI researchers, senior engineering leadership, design, and product teams to translate AI capabilities into user-facing features.
• Lead strategic initiatives for application feature enhancements through advanced AI-driven approaches.
• Collaborate closely with legal and compliance stakeholders to ensure responsible and explainable AI outcomes.

Benefits

• Hybrid Work Model: Flexible hybrid working environment (2-3 days a week in the office).
• Flexibility & Work-Life Balance: Supportive workplace policies for personal and professional responsibilities.
• Career Development and Growth: Continuous learning and skill development opportunities.
• Industry Competitive Benefits: Comprehensive benefit plans including flexible vacation, mental health days, retirement savings, and tuition reimbursement.
• Culture: Award-winning reputation for inclusion and belonging.
• Social Impact: Opportunities for community involvement and pro-bono consulting projects.
• Making a Real-World Impact: Helping customers pursue justice, truth, and transparency.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: Bachelor's degree