Staff AI Engineer: Agentic Systems & Geospatial Reasoning

Scaleai; Denver (CO); On-site; USD 274,000 - 343,000; Full time

Benefits offered by this job

Health, dental and vision coverage

Retirement benefits

Learning and development stipend

Generous PTO

Commuter stipend

Job summary

Scale is seeking a senior staff ML architect to lead the design and deployment of agentic AI systems in production environments. You will guide end-to-end systems, including long-horizon reasoning, geospatial data interpretation, and robust retrieval pipelines.

You will collaborate with managers, researchers, and product leaders to drive high-impact initiatives, mentor teams, and ensure production readiness for stateful AI deployments in mission-critical contexts.

Qualifications

• 8+ years of experience building and deploying applied ML systems in production environments.
• Deep experience with agentic systems, autonomous workflows, or ML systems that reason and act over multiple steps.
• Strong background in ML systems engineering, including model serving, pipelines, monitoring, and evaluation.
• Hands-on experience with retrieval systems, embeddings, or representation learning.
• Proficiency in Python and modern ML frameworks (ex: PyTorch), with the ability to design systems end to end.
• Demonstrated ability to operate at Staff-level scope: setting technical direction, owning ambiguous problems, and driving 0→1 initiatives to production.
• Experience making thoughtful tradeoffs across performance, cost, reliability, and development velocity.

Responsibilities

• Lead the architecture and implementation of agentic AI systems, focusing on long-horizon reasoning, orchestration, and system-level reliability.
• Build and scale agents that perform complex geospatial reasoning, including maps, spatial data interpretation and generation.
• Design and improve retrieval systems across large collections of documents for high-signal context.
• Fine-tune and evaluate embedding models to improve recall and precision for critical datasets.
• Design memory systems enabling persistent state over long contexts and learning from interactions.
• Own and evolve shared agentic infrastructure and core libraries for reuse across teams and contracts.
• Define evaluation strategies for robustness, failure modes, and regression testing in production.
• Partner with engineering managers, product leaders, and researchers to scope high-impact initiatives.
• Serve as a technical mentor raising the bar for design, ML rigor, and production readiness.
• Comfortable with light travel (~10%) for customer interaction and team needs.

Skills

Agentic AI systems

Geospatial reasoning

Retrieval systems

Embedding models

Python

ML frameworks

Staff-level leadership

System design

Tools

Model serving

Monitoring

Job description

Scale is seeking a senior staff ML architect to lead the design and deployment of agentic AI systems in production environments. You will guide end-to-end systems, including long-horizon reasoning, geospatial data interpretation, and robust retrieval pipelines.

You will collaborate with managers, researchers, and product leaders to drive high-impact initiatives, mentor teams, and ensure production readiness for stateful AI deployments in mission-critical contexts.