Staff/Principal AI Engineer
Leads the technical AI strategy and production application of LLMs, NLP, retrieval, agents, and multimodal systems for large-scale legal discovery. Requires 8+ years of software engineering experience, including significant ML/AI leadership and production LLM experience.
About the job
Responsibilities
- Lead the design and implementation of Everlaw's technical AI strategy.
- Evaluate frontier models, open-source offerings, and evolving AI APIs, including participation in early-access programs and providing feedback on model and API design.
- Develop approaches for applying AI at trillion-token scale to complex litigation reasoning tasks and multimodal inputs.
- Guide engineers across the company in applying AI through design-document feedback, code reviews, technical talks, and study groups.
- Collaborate with software engineers, security engineers, data scientists, product leads, designers, and other cross-functional teams.
- Identify opportunities to apply advances in ML and AI to legal technology and influence engineering and product direction.
Requirements
- At least 6 years of experience in ML/AI and 8+ years of overall software engineering experience.
- Experience leading significant AI projects, including applying LLMs in production.
- Experience with NLP and text analysis.
- Programming skills in at least one scripting language, such as Python.
- Demonstrated ability to stay current with AI developments and apply them to engineering work.
- Strong cross-functional collaboration and communication skills, including the ability to explain technical concepts clearly and mentor others.
Nice-to-haves
- Experience with retrieval-based language models or hybrid RAG approaches for large custom datasets.
- Experience designing knowledge storage and representation approaches.
- Familiarity with vector databases and graph databases.
- Experience with multimodal processing, including images, audio, or video.
- Experience building task-oriented agents, including supervision, backtracking, exception handling, tool use, or MCP.
- Experience with SaaS and distributed systems.
Compensation and Benefits
- Expected salary range: $228,000–$288,000.
- Equity program.
- 401(k) retirement plan with company matching.
- Health, dental, and vision insurance.
- Flexible Spending Accounts for health and dependent care expenses.
- Paid parental leave and approximately 10 days (80 hours) of sick leave.
- Seventeen paid vacation days plus 11 federal holidays.
- Modern Health membership for mental health and wellness.
- Annual learning and development allocation and applicable professional membership dues.
- Company-sponsored life and disability insurance.
- Onsite position in Oakland, California, with 3 days per week in the office.
Skills
Artificial Intelligence, Machine Learning, LLMs, NLP, Text Analysis, Python, RAG, Vector Databases, Graph Databases, Multimodal Ai, AI Agents, Mcp, SaaS, Distributed Systems
Similar jobs
ML Engineering jobs
San Francisco, CA
Principal technical leader defining architecture and multi-year strategy for Pinterest’s Homefeed, Search, and AI Assistant experiences. The role requires 15+ years of large-scale systems or machine-learning experience, deep expertise in discovery and generative AI, and hands-on leadership across engineering and product organizations.
PointClickCare
United States
Leads the technical vision, architecture, and engineering standards for a company-wide ML platform supporting model development, deployment, serving, and monitoring. The role requires principal-level expertise in Python and Java, scalable MLOps, cloud infrastructure, security, and technical leadership across teams.
Rula
United States
Own the architecture and delivery of production AI systems for patient-provider matching, search relevance, personalization, clinical workflows, and engagement. The role requires extensive software engineering, distributed systems, search or recommendation, ML applications, and foundation-model experience in a regulated healthcare setting.
Zoox
Foster City, CA
Leads the technical strategy and engineering execution required to achieve driverless freeway operation for autonomous vehicles. Requires 10+ years of software experience, demonstrated freeway autonomy leadership, deep expertise in an autonomy domain, and strong executive communication skills.
Fetch
United States
Design and scale ML infrastructure and real-time learning systems powering personalization, search, ranking, and ad tech for millions of consumers. The role requires deep distributed-systems and data-pipeline expertise, strong architecture leadership, and experience delivering zero-to-one ML systems.