AI Engineer
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 job
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 Learning, Fine-Tuning, LLMs, Ai Reasoning, Model Optimization, Evaluation Frameworks, Python, PyTorch, Transformers, Prompt Engineering
Similar jobs
ML Engineering jobs
Nuro
Mountain View, CABuild trustworthy infrastructure for production LLM agents, closed-loop evaluation, and autonomous research workflows. The role requires strong Python and distributed-systems experience, hands-on LLM post-training and inference knowledge, and experience operating agent systems at scale.
ClickUp
United States
Build and operate large-scale ranking and retrieval systems that power search relevance, including hybrid lexical/vector search, embeddings, query understanding, and permission-aware retrieval. Requires a bachelor's degree and 5+ years of ML engineering experience in ranking or information retrieval.
Atomicmachines
Emeryville, CA
Build and operate production ML infrastructure spanning training, deployment, serving, monitoring, data pipelines, and feedback-driven retraining. The role requires strong MLOps and DevOps experience, Python and SQL proficiency, and ownership of reliable cloud-based systems.
Tessera Labs
San Jose, CABuild and scale post-training, reinforcement-learning, evaluation, and inference systems for long-horizon agents operating over complex enterprise software. The role requires strong Python and PyTorch or JAX skills, distributed GPU experience, empirical rigor, and the ability to take research results into production.
Tessera Labs
San Jose, CABuild and operate production AI agents that transform enterprise processes, data, and code. The role focuses on tool layers, retrieval, context management, evaluations, monitoring, auditability, and guardrails, requiring strong Python and TypeScript plus experience with production LLM systems and traditional machine learning.