Staff ML/AI Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Everlaw · 9 hours ago

Staff ML/AI Engineer

Everlaw is looking for a Staff ML/AI Engineer to help build intelligent systems for legal professionals. In this role, you will design architecture, mentor engineers, and apply ML/AI to enhance the company's litigation technology products.

Cloud ComputingDocument ManagementLegalLegal Tech
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Apply core ML Fundamentals: Apply a deep understanding of probability, statistics, and optimization to ensure our models are not just "smart," but robust and explainable
Lead the design and implementation of our ML infrastructure and tooling. Champion MLOps best practices, ensuring that our experimentation-to-production lifecycle is efficient and reproducible
Help elevate the technical bar of the engineering team. We’re an organization devoted to constant learning. As a staff engineer, you’ll lead by example in staying abreast of industry and academic research developments and sharing your learnings with the team – whether in structured ways like feedback on design documents or code reviews, or in less formal ways, like giving occasional tech talks or leading a study group on ML research papers
Collaborate with other software engineers, security engineers, data scientists, product leads and designers, and other cross-functional teams in the pursuit of our goal of being the AI leader in litigation
Be on the lookout for new opportunities that ML and AI enable. Help us spot cases where industry or research progression in ML/AI tech will translate to Legal Tech, and influence our engineering and product direction. We’re prepared to act and innovate in bringing the best ideas to market

Qualification

ML FundamentalsMLOpsCloud InfrastructureProduction SaaS ExperiencePythonPyTorchTensorFlowCollaborationMentoringContinuous Learning

Required

8+ years of experience in software engineering, with at least 4 years dedicated to building and deploying ML models in a production SaaS environment
A solid grounding in ML Fundamentals: This includes command of the math behind the models (linear algebra, calculus, and loss function optimization)
Emphasize practicality and product outcomes over hype: You'll know when a simple random forest or Regressor will do the job vs. a multi-billion parameter model and Transformers
A track record of solving 'noisy data' problems and understanding precision/recall tradeoffs. You'll help teams design evaluation strategies to achieve their product goals
MLOps experience, making use of cloud-based infrastructure and technologies like Kubernetes, Docker, Weights & Biases, MLFlow, etc and major cloud providers like AWS, GCP, or Azure. You've designed systems with reliability and observability in mind
Proficient in at least one language (like Python, Go, or Java) and experienced with frameworks such as PyTorch, TensorFlow, or JAX. You write production-grade code that is clean, tested, and scalable
A constant learner, staying abreast of the latest ML/AI developments from the industry, and how you might apply these to your work
Able to collaborate effectively with coworkers on different teams. You can zoom in to discuss technical details with other experienced AI engineers, and can zoom out to explain technical concepts without jargon. You're willing to share your knowledge and help others learn

Preferred

You've applied LLMs to production-grade software, including knowledge of fine-tuning, prompt engineering, and familiarity with the latest offerings from both the frontier labs and open source
You've devised embedding strategies and approaches for knowledge representation, including vector and graph databases and applied RAG to find answers in very large custom data sets
You've built agents, MCP servers or clients, or have designed evaluation strategies for monitoring and supervising AI agents

Benefits

Equity program
401(k) retirement plan with company matching
Health, dental, and vision
Flexible Spending Accounts for health and dependent care expenses
Paid parental leave and approximately 10 days (80 hours) per year of sick leave
Seventeen paid vacation days plus 11 federal holidays
Membership to Modern Health to help employees prioritize mental health and wellness
Annual allocation for Learning & Development opportunities and applicable professional membership dues
Company-sponsored life and disability insurance
Time off for company-sponsored volunteer events and 4 paid hours per quarter to volunteer at a charitable organization of your choice

Company

Everlaw offers a platform for document analysis to law firms, government, and corporations.

H1B Sponsorship

Everlaw has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (8)
2024 (11)
2023 (9)
2022 (5)
2021 (2)

Funding

Current Stage
Late Stage
Total Funding
$298.6M
Key Investors
TPG GrowthCapitalGMenlo Ventures
2021-11-02Series D· $202M
2020-03-10Series C· $62M
2018-06-28Series B· $25M

Leadership Team

leader-logo
AJ Shankar
Founder, CEO
linkedin
leader-logo
Scott Allen
Chief Financial Officer
linkedin
Company data provided by crunchbase