Skip to content

AI Engineer

195k – 300kSan Mateo, CAML EngineeringHybrid
Summary

Build and deploy AI systems for legal workflows, fine-tuning models on domain-specific data, advancing reasoning capabilities, and integrating with product features. Requires Bachelor's/Master's in ML/AI/CS and proven production ML experience.

About the role

What You'll Do

  • Develop AI-Powered Solutions: Design and implement AI systems that understand, analyze, and provide intelligent assistance across various legal tasks and queries.
  • Optimize and Integrate Models: Fine-tune models on high-quality, domain-specific data (code, natural language, product usage signals).
  • Advance AI Reasoning: Improve the reasoning capabilities of our AI systems to enhance decision-making and support complex legal domains.
  • Collaboration: Work closely with product engineers to integrate models into real user-facing features. Partner closely with legal professionals to translate their expertise into technical requirements, tailoring AI solutions to legal industry needs.
  • Build Rigorous Evaluation Frameworks: Establish frameworks that rigorously measure and evaluate AI performance, ensuring meaningful, reliable outcomes.

What We're Looking For

  • Educational Background: Bachelor’s or Master’s degree in Machine Learning, Artificial Intelligence, Computer Science, or a related field.
  • Experience in Complex AI Systems: Proven experience developing advanced AI systems, particularly those that deconstruct complex tasks into manageable steps. Shipped ML systems in production, with real users and high uptime.
  • Expertise in Model Optimization: Proficiency in optimizing and fine-tuning foundational models, with a desire to push AI capabilities forward.
  • Analytical Skills: Strong ability to measure impactful metrics and build frameworks to assess AI performance effectively.
  • Problem-Solving Skills: Skilled in integrating various models and tools to create robust solutions for real-world applications.

Compensation and Benefits

US Base Salary Range: $195,000—$300,000 USD

  • Competitive Salary & Equity
  • 401(k) Program with Employer Matching
  • Health, Dental, Vision and Life Insurance
  • Short Term and Long Term Disability
  • Commuter Benefits*
  • Autonomous Work Environment
  • Workplace Setup Reimbursement
  • Telecomm Stipend
  • Flexible Time Off (FTO) + Holidays
  • Quarterly Team Gatherings
  • In office Perks*

*In office employees only

Skills
Machine LearningFine-tuningLLMsAI ReasoningModel OptimizationEvaluation FrameworksPythonPyTorchTransformersPrompt Engineering
Similar roles at this salary range
All ML Engineering jobs →
Mem0

Senior Research Engineer

Own the end-to-end lifecycle of memory features for AI agents. Fine-tune models, implement research, build evaluations, and ship production systems with Engineering.

175k – 250kSan Francisco, CAML EngineeringOn-site7+ YOERAGvLLM
Ironclad

Senior Software Engineer, AI

Lead design and delivery of high-priority AI initiatives across multiple codebases. Build and ship AI-powered features with strong backend fundamentals and product sense.

180k – 220kSan Francisco, CAML EngineeringHybrid5+ YOEReactEvals
Mercury

Senior Machine Learning Operations Engineer

Build and operate Mercury's real-time ML inference platform for fraud risk decisioning. Own model deployment, observability, and lifecycle tooling with strong backend Python fundamentals.

167k – 208kSan Francisco, CA +2ML EngineeringHybrid5+ YOESQLSHAP
Plaid

Machine Learning Engineer - Embedded Insights

Drive ML initiatives from concept to production on the Embedded Insights team. Identify opportunities, build and deploy models using Plaid's financial datasets, and partner with product teams to deliver scalable customer-facing intelligence products.

212k – 272kSan Francisco, CA +2ML EngineeringHybrid5+ YOESQLMLOps
Plaid

Machine Learning Engineer

Advance Plaid’s foundation models by developing novel architectures, pretraining objectives, and fine-tuning strategies. Work across the full ML stack from data engineering to production serving and monitoring.

212k – 272kSan Francisco, CA +2ML EngineeringHybrid1+ YOELLMsPython