Lead AI Engineer
Leads development of proprietary AI reasoning model TRAM for interpreting global trade law, building data pipelines, fine-tuning LLMs, and evaluation frameworks for high-speed, accurate compliance determinations. Requires AI product experience, especially RAG systems and model fine-tuning.
About the job
What You'll Do
Within weeks:
- Lead development of new features aimed at increasing TRAM’s test-time accuracy
- Work on the underlying data and retrieval pipelines that help power our AI workflows
- Work directly with our internal tax experts to understand how TRAM can better reason like them
Within months:
- Own TRAM’s eval framework and workflows
- Work directly with leading frontier labs to reinforce fine tune models on our proprietary data
Requirements
- Prior experience building AI enabled products, particularly RAG systems
- Experience fine tuning base models, ideally via RF
- Willingness to dive into tax technical problems
- A strong understanding of how LLMs and reasoning models function
Nice to Haves
- Experience working with LLMs on legal applications
- Experience with RAG data pipelines and collecting/curating data for the pipeline
Skills
RAG, LLMs, Fine-Tuning, Retrieval Pipelines, Evaluation Frameworks, Data Pipelines, Reasoning Models
Similar jobs
ML Engineering jobs
Baselayer
San Francisco, CA
Senior AI Engineer responsible for production LLM agents that enrich business identity data through web discovery, verification, classification, and risk scoring. The role requires strong asynchronous Python, agent and evaluation expertise, browser automation, and experience operating AI systems in production.
Otter
Mountain View, CA
Leads the development and production deployment of large-scale ASR and TTS systems for conversational intelligence products. The role requires 5+ years of industry experience, deep speech-model expertise, and strong software engineering and ML operations capabilities.
Discord
United States
Leads Discord’s Safety ML team, setting technical direction and overseeing production machine learning systems for content understanding, account integrity, and platform abuse. Requires substantial machine learning and engineering management experience, hands-on technical depth, and experience delivering ML systems at scale.
Ambience Healthcare
San Francisco, CA
Build and improve production AI systems for clinical products, owning evaluations, model behavior, agentic workflows, data flywheels, deployment, and observability. The role requires 5+ years of production ML or applied AI experience, strong Python and modern ML framework skills, and hands-on debugging expertise.
Dialpad
United States
Leads development of speech models, decoders, and low-latency inference systems for next-generation voice agents. Requires 5+ years in speech ML or related audio AI, strong Python and PyTorch experience, and the ability to guide technical direction and mentor engineers.