Staff Machine Learning Engineer, Public Sector
AI summary of the role
Staff Machine Learning Engineer leading architecture and production deployment of agentic AI systems for classified and regulated government environments.
What you’ll do
- Lead architecture and implementation of reliable agentic AI systems
- Build agents for complex geospatial reasoning over maps and spatial data
- Design retrieval, embedding, and memory systems for mission-critical datasets
- Own shared agentic infrastructure and evaluation strategies
What you’ll bring
- 8+ years building and deploying applied ML systems in production
- Deep experience with agentic systems or multi-step autonomous workflows
- Strong ML systems engineering background spanning serving, pipelines, monitoring, and evaluation
- Hands-on retrieval, embeddings, or representation learning experience
Technologies
Python · PyTorch · LoRA/PEFT · RLHF · LLM-as-judge · GEOINT · embedding models · agentic AI
About Scale AI
Data, evals, and GenAI platform for frontier labs, enterprises, and governments — the picks-and-shovels layer for production AI.
Series G
Employer postings · Data from · Sources