Oakland, California, United States Hybrid USD 228,000 - 288,000 Yearly

Everlaw is hiring a Staff ML/AI Engineer

Everlaw is hiring a Staff ML/AI Engineer to build the intelligent systems that help legal professionals navigate complex litigation cases at the scale of millions of documents. This role bridges the gap between cutting-edge research and production-grade software, bringing new capabilities to market. You will have a key role in designing architecture, mentoring senior engineers, and defining how ML/AI are applied in our product.

What You'll Do

  • Apply a deep understanding of probability, statistics, and optimization to ensure models are robust and explainable.
  • Lead the design and implementation of ML infrastructure and tooling, championing MLOps best practices.
  • Help elevate the technical bar of the engineering team by staying abreast of industry research and sharing learnings.
  • Collaborate with software engineers, security engineers, data scientists, product leads, and other cross-functional teams.
  • Be on the lookout for new opportunities that ML and AI enable, influencing engineering and product direction.

What We're Looking For

  • 8+ years of experience in software engineering.
  • At least 4 years dedicated to building and deploying ML models in a production SaaS environment.
  • Solid grounding in ML Fundamentals including command of the math behind the models (linear algebra, calculus, loss function optimization).
  • Emphasis on practicality and product outcomes over hype.
  • Track record of solving 'noisy data' problems and understanding precision/recall tradeoffs.
  • MLOps experience with cloud-based infrastructure and technologies like Kubernetes, Docker, Weights & Biases, MLFlow, etc.
  • Proficient in at least one language (like Python, Go, or Java) and experienced with frameworks such as PyTorch, TensorFlow, or JAX.
  • Constant learner, staying abreast of the latest ML/AI developments.
  • Able to collaborate effectively with coworkers on different teams and explain technical concepts without jargon.

Nice to Have

  • Applied LLMs to production-grade software, including knowledge of fine-tuning, prompt engineering.
  • Devised embedding strategies and approaches for knowledge representation, including vector and graph databases and applied RAG.
  • Built agents, MCP servers or clients, or have designed evaluation strategies for monitoring and supervising AI agents.

Technical Stack

  • Languages: Python, Go, Java
  • Frameworks: PyTorch, TensorFlow, JAX
  • Infrastructure: Kubernetes, Docker, Weights & Biases, MLFlow
  • Cloud Platforms: AWS, GCP, Azure

Team & Environment

You will collaborate with a community of other senior staff and principal engineers. The role has regular visibility to, and interactions with, the Chief Technology Officer and other senior leaders.

Benefits & Compensation

  • Compensation: $228,000 - $288,000 + equity program
  • 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
  • Approximately 10 days (80 hours) per year of sick leave
  • Seventeen paid vacation days plus 11 federal holidays
  • Membership to Modern Health for mental health and wellness
  • Annual allocation for Learning & Development opportunities and professional membership dues
  • Company-sponsored life and disability insurance

Work Mode

This is a hybrid role based in Oakland, California.

Everlaw is an equal opportunity employer. We pride ourselves on having a diverse workforce and we do not discriminate against any employee or applicant because of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition, or any other basis protected by law.

Required Skills
PythonGoJavaPyTorchTensorFlowJAXKubernetesDockerWeights & BiasesMLFlowMachine LearningSaaSLinear AlgebraCalculusOptimization
Want to work from Thailand?

Join a remote network built for tech talent

Iglu gives you real employment in Southeast Asia — visa, work permit, and projects included. Pick what you work on, earn performance-based pay, and live where you want.

Legal employment in Thailand & Vietnam
Choose your own projects
Performance-based revenue sharing
Relocation support available
Join Iglu
200+ professionals worldwide
About company
Everlaw
Everlaw helps law firms, government agencies, and corporations sift through millions of documents of evidence in big lawsuits and investigations. It's a multi-billion dollar space typically dominated by service-oriented vendors, and Everlaw comes at it with cutting-edge technology and elegant design.
All jobs at Everlaw Visit website
Job Details
Department Software Development
Category data
Posted 3 months ago