Staff Machine Learning Engineer, Public Sector

📋 Description

• Lead architecture of agentic AI systems for long-horizon reasoning and reliability.
• Build and scale agents for geospatial reasoning over maps and spatial data.
• Design and improve retrieval systems across large document collections.
• Fine-tune and evaluate embedding models to boost recall and precision.
• Design memory systems to persist state and handle long contexts.
• Own and evolve shared agentic infrastructure for reuse across teams; clearance required or obtainable.

🎯 Requirements

• 8+ years of experience building and deploying applied ML systems in production.
• Deep experience with agentic systems or ML systems that reason over multiple steps.
• Strong background in ML systems engineering: model serving, pipelines, monitoring, evaluation.
• Hands-on experience with retrieval systems, embeddings, or representation learning.
• Proficiency in Python and modern ML frameworks (e.g., PyTorch) for end-to-end design.
• Staff-level: set technical direction, own ambiguous problems, drive 0→1 initiatives to production.

🎁 Benefits

• Health, dental, and vision coverage
• Retirement benefits
• Learning and development stipend
• Generous PTO
• Commuter stipend
• Equity-based compensation eligibility