Skip to content
SphereSphereSan Francisco, CA

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.

250k – 280k/yr
On-siteML Engineering

About the role

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

RAGLLMsFine-TuningRetrieval PipelinesEvaluation FrameworksData PipelinesReasoning Models

Similar roles

ML Engineering jobs
Fireworks AI

MTS, Research Engineer

Fireworks AINew York, NY +1

Research Engineer conducting open-ended ML research, reproducing SOTA papers, building/scaling distributed training infrastructure on GPU clusters, and bridging research ideas into production code. Requires strong programming, ML frameworks, distributed systems experience, and math foundations; Master's/PhD preferred.

250k – 400k/yr
On-site7+ YOEML Engineering
Decagon

Senior Software Engineer, Agents

DecagonNew York, NY

Build and iterate on AI agents for customer interactions, outperforming humans in complex scenarios. Requires 5+ years experience with Python/TypeScript, systems debugging, and model evaluation/integration.

250k – 330k/yr
On-site5+ YOEML Engineering
Decagon

Senior Software Engineer, Agents

DecagonSan Francisco, CA

Build and deploy scalable AI agents for customer support that handle complex interactions across industries like finance and healthcare. Requires 5+ years experience with Python and TypeScript, focusing on system reliability and model integration.

250k – 330k/yr
On-site5+ YOEML Engineering
Scale AI

Tech Lead Manager- MLRE, ML Systems

Scale AISan Francisco, CA +2

Leads development and optimization of distributed frameworks for LLM post-training, training, and inference. Collaborates with ML teams to enable advanced model development and data curation, requiring expertise in large-scale ML systems and tools like PyTorch and CUDA.

252k – 315k/yr
On-siteML Engineering
Garner Health

Senior Machine Learning Operations Engineer

Garner HealthNew York City, NY

Build and operate production ML systems and platform components for healthcare technology, partnering with ML and data science teams on model deployment, observability, and reliability.

256k – 285k/yr
Hybrid5+ YOEML Engineering