AI & ML Engineer
SandstoneLegal Workflow company
New York, United StatesSenior
Lightspeed Venture Partners
Sequoia Capital
SV Angel
Mantis VC
Operator Partners
Kearny Jackson
Data & AI
About the role
TL;DR
Build production AI systems for legal workflows, focusing on LLM-powered applications.
- •Sandstone is building the AI-native operating system for in-house legal teams.
- •As an AI & ML Engineer, you will build the systems that make that possible: agents that reason over contracts and business context, retrieval systems that surface the right precedent, document understanding pipelines that turn legal work into structured intelligence, and evals that help us measure and improve quality.
- •Key Responsibilities Build production AI systems for legal workflows such as contract review, redlining, legal intake, drafting, negotiation, search, and knowledge management Improve retrieval and context systems that reason over contracts, playbooks, precedent, customer data, and business context Build document parsing and understanding pipelines that extract structure, entities, obligations, issues, and negotiation context from complex legal documents Requirements 4+ years of experience as an AI engineer, ML engineer, full-stack engineer, product engineer, data scientist, or similar role Experience shipping applied AI or LLM-powered products, especially involving retrieval, agents, evals, document understanding, structured extraction, or workflow orchestration Strong software engineering fundamentals and comfort working across backend systems, data pipelines, APIs, and product-facing surfaces
Required skills
PythonLLMsLangChainNLPDockerKubernetesCI/CDGitLinux
Domain expertise
legaltechaiproductivity
Benefits & perks
Comprehensive health, dental, and vision insurance, 401K, 12 weeks paid parental leave, Flexible time off (20+ days PTO), Annual learning & development stipend, Technology stipend, including unlimited spend on AI tooling, Daily lunch for in-office team members
Tech stack
PythonJavaScriptTypeScriptSQLBashReactNode.jsDjangoFlaskFastAPIPandasNumPySparkAirflowTensorFlowPyTorchscikit-learnKerasOpenAI APIAnthropic APIGemini APILangChainHugging FaceNLPLLMsMLflowKubeflowSageMakerVertex AIPostgreSQL