Staff Software Engineer (AI Platform)
ΠΡΡΡ & Π‘ΠΎΠΏΡΠΎΠ²ΠΎΠ΄
ΠΠ»Ρ ΠΌΡΡΡΠ° Ρ ΡΡΠΎΠΉ Π²Π°ΠΊΠ°Π½ΡΠΈΠ΅ΠΉ Π½ΡΠΆΠ΅Π½ Plus
ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ Π²Π°ΠΊΠ°Π½ΡΠΈΠΈ
TL;DR
Staff Software Engineer (AI Platform): Building and optimizing the model layer and agent infrastructure for a legal AI platform with an accent on context management, model routing, and evaluation frameworks. Focus on designing shared abstractions and SDKs to enable rapid deployment of autonomous AI systems in high-stakes environments.
Location: San Francisco
Salary: $231,000 β $340,000 USD
Company
is transforming legal and professional services by combining frontier agentic AI with an enterprise-grade platform to reshape critical knowledge work.
What you will do
- Design and build platform-level abstractions and systems that improve all agentic products.
- Own infrastructure for model integration, routing, and evaluation to deploy the right foundation models.
- Develop shared evaluation frameworks and tooling to systematically measure and improve AI quality.
- Partner with product engineering, PMs, and design to launch cutting-edge AI products.
- Evaluate, prototype, and integrate the latest advancements in AI and agentic systems.
Requirements
- 8+ years of experience building backend systems.
- 1+ year of focused experience in AI/ML engineering.
- Proven track record of technical leadership across multiple teams.
- Experience shipping multi-model or multi-provider AI systems in production.
- Familiarity with context management, session state, or memory systems in AI or distributed systems.
- Experience building internal platforms or SDKs adopted by other engineering teams.
Nice to have
- Experience building evaluation frameworks.
- Experience working with agent or function-calling architectures.
- Familiarity with legal or other high-stakes professional services domains.
- Experience at early-stage or hyper-growth startups.
Culture & Benefits
- Opportunity to build a generational company at a true inflection point.
- Fast-paced environment emphasizing ownership, intensity, and excellence.
- Culture driven by the values of Decisiveness, Simplicity, and a "Job's Not Finished" mindset.
- Exposure to a scaling product with 1500+ customers and world-class investor support.
ΠΡΠ΄ΡΡΠ΅ ΠΎΡΡΠΎΡΠΎΠΆΠ½Ρ: Π΅ΡΠ»ΠΈ ΡΠ°Π±ΠΎΡΠΎΠ΄Π°ΡΠ΅Π»Ρ ΠΏΡΠΎΡΠΈΡ Π²ΠΎΠΉΡΠΈ Π² ΠΈΡ ΡΠΈΡΡΠ΅ΠΌΡ, ΠΈΡΠΏΠΎΠ»ΡΠ·ΡΡ iCloud/Google, ΠΏΡΠΈΡΠ»Π°ΡΡ ΠΊΠΎΠ΄/ΠΏΠ°ΡΠΎΠ»Ρ, Π·Π°ΠΏΡΡΡΠΈΡΡ ΠΊΠΎΠ΄/ΠΠ, Π½Π΅ Π΄Π΅Π»Π°ΠΉΡΠ΅ ΡΡΠΎΠ³ΠΎ - ΡΡΠΎ ΠΌΠΎΡΠ΅Π½Π½ΠΈΠΊΠΈ. ΠΠ±ΡΠ·Π°ΡΠ΅Π»ΡΠ½ΠΎ ΠΆΠΌΠΈΡΠ΅ "ΠΠΎΠΆΠ°Π»ΠΎΠ²Π°ΡΡΡΡ" ΠΈΠ»ΠΈ ΠΏΠΈΡΠΈΡΠ΅ Π² ΠΏΠΎΠ΄Π΄Π΅ΡΠΆΠΊΡ. ΠΠΎΠ΄ΡΠΎΠ±Π½Π΅Π΅ Π² Π³Π°ΠΉΠ΄Π΅ β