Staff Machine Learning Engineer, Public Sector

Job summary

Responsibilities

The goal of a Staff Machine Learning Engineer at Scale is to lead the design and deployment of agentic AI systems that operate in real-world, mission-critical government environments.

• Lead the architecture and implementation of agentic AI systems, with a focus on long-horizon reasoning, orchestration, and system-level reliability.
• Build and scale agents that perform complex geospatial reasoning, including interpreting, generating, and reasoning over maps and spatial data.
• Design and improve retrieval systems across large collections of static and semi-structured documents, enabling agents to surface high-signal context efficiently.
• Fine-tune and evaluate embedding models to improve recall and precision for mission-critical datasets.
• Design memory systems that allow agents to persist state, operate over long contexts, and learn from prior interactions.

Tech stack

• Python
• PyTorch