Staff Machine Learning Engineer, Public Sector Job at DiversityJobs Inc in New Y

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. On the Public Sector team, you'll work at the intersection of agentic ML, systems engineering, and applied research, building foundational infrastructure that enables AI systems to reason, plan, and act reliably at national scale.

Our Public Sector ML Team partners directly with U.S. defense and intelligence agencies to deploy AI into classified and regulated environments. Through flagship programs like

Donovan

and

Thunderforge , we are advancing the next generation of agentic AI for geospatial reasoning, planning, and decision support. Staff Machine Learning Engineers play a central role in setting technical direction, owning core architectures, and translating ambitious ideas into production systems trusted by government operators.

You will:

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.

Own and evolve shared agentic infrastructure and core libraries, enabling reuse across teams, products, and Public Sector contracts.

Define evaluation strategies for agentic systems, including robustness testing, failure-mode analysis, and regression testing in production environments.

Partner closely with engineering managers, product leaders, and researchers to scope high-impact initiatives and unblock execution across teams.

Serve as a technical mentor and multiplier-raising the bar for system design, ML rigor, and production readiness across the organization.

This role will require an active security clearance or the ability to obtain a security clearance.

Ideally You'd Have:

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 01 initiatives to production.

Experience making thoughtful tradeoffs across performance, cost, reliability, and development velocity.

Nice to Haves:

High ownership over 01 systems that move directly into production.

Real-world constraints that force thoughtful engineering tradeoffs, not just model tuning.

Opportunity to shape foundational agentic infrastructure used across multiple teams and missions.

Work that blends research depth with applied impact, in environments where correctness, robustness, and trust matter.

Compensation packages at Scale for eligible roles include base salary, equity, and benefits. The range displayed on each job posting reflects the minimum and maximum target for new hire salaries for the position, determined by work location and additional factors, including job-related skills, experience, interview performance, and relevant education or training. Scale employees in eligible roles are also granted equity based compensation, subject to Board of Director approval. Your recruiter can share more about the specific salary range for your preferred location during the hiring process, and confirm whether the hired role will be eligible for equity grant. You'll also receive benefits including, but not limited to: Comprehensive health, dental and vision coverage, retirement benefits, a learning and development stipend, and generous PTO. Additionally, this role may be eligible for additional benefits such as a commuter stipend.
The base salary range for this full-time position in the location of Washington DC is:

$260,400

—

$326,600 USD

Compensation packages at Scale for eligible roles include base salary, equity, and benefits. The range displayed on each job posting reflects the minimum and maximum target for new hire salaries for the position, determined by work location and additional factors, including job-related skills, experience, interview performance, and relevant education or training. Scale employees in eligible roles are also granted equity based compensation, subject to Board of Director approval. Your recruiter can share more about the specific salary range for your preferred location during the hiring process, and confirm whether the hired role will be eligible for equity grant. You'll also receive benefits including, but not limited to: Comprehensive health, dental and vision coverage, retirement benefits, a learning and development stipend, and generous PTO. Additionally, this role may be eligible for additional benefits such as a commuter stipend.
Please reference the job posting's subtitle for where this position will be located. For pay transparency purposes, the base salary range for this full-time position in the locations of San Francisco, New York, Seattle is:

$289,800

—

$362,250 USD

Please reference the job posting's subtitle for where this position will be located. For pay transparency purposes, the base salary range for this full-time position in the locations of Washington DC, Texas, Colorado, Hawaii is:

$260,400

—

$326,600 USD

PLEASE NOTE:

Our policy requires a 90-day waiting period before reconsidering candidates for the same role. This allows us to ensure a fair and thorough evaluation of all applicants.

About Us:

At Scale, our mission is to develop reliable AI systems for the world's most important decisions. Our products provide the high-quality data and full-stack technologies that power the world's leading models, and help enterprises and governments build, deploy, and oversee AI applications that deliver real impact. We work closely with industry leaders like Meta, Cisco, DLA Piper, Mayo Clinic, Time Inc., the Government of Qatar, and U.S. government agencies including the Army and Air Force. We are expanding our team to accelerate the development of AI applications.

We believe that everyone should be able to bring their whole selves to work, which is why we are proud to be an inclusive and equal opportunity workplace. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability status, gender identity or Veteran status.

We are committed to working with and providing reasonable accommodations to applicants with physical and mental disabilities. If you need assistance and/or a reasonable accommodation in the application or recruiting process due to a disability, please contact us at accommodations@scale.com. Please see the United States Department of Labor's

Know Your Rights poster

for additional information.

We comply with the United States Department of Labor's

Pay Transparency provision

.

PLEASE NOTE:

We collect, retain and use personal data for our professional business purposes, including notifying you of job opportunities that may be of interest and sharing with our affiliates. We limit the personal data we collect to that which we believe is appropriate and necessary to manage applicants' needs, provide our services, and comply with applicable laws. Any information we collect in connection with your application will be treated in accordance with our internal policies and programs designed to protect personal data. Please see our

privacy policy

for additional information.

In Summary: 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 . The Public Sector ML Team partners directly with U.S. defense and intelligence agencies to deploy AI into classified environments .

En Español:

El objetivo de un ingeniero de aprendizaje automático del personal a escala es liderar el diseño y la implementación de sistemas de IA agenciales que operan en entornos gubernamentales reales, críticos para las misiones. En el equipo del sector público, trabajará en la intersección de AI agencial, ingeniería de sistemas e investigación aplicada, construyendo una infraestructura fundamental que permita a los sistemas de AI razonar, planificar y actuar confiablemente a escala nacional. Nuestro Equipo ML del Sector Público se asocia directamente con las agencias de defensa y inteligencia de Estados Unidos para implementar AI en ambientes clasificados y regulados. Los ingenieros de aprendizaje automático del personal desempeñan un papel central en establecer la dirección técnica, poseer arquitecturas centrales y configurar datos espaciales. Diseñar e impulsar sistemas de recuperación a través de grandes colecciones de documentos estáticos y semistruturados que sean confiados por los operadores gubernamentales. Dirigirás: liderar la arquitectura e implementación de sistemas de IA agenciales, con enfoque en el razonamiento a largo horizonte, orquestación y fiabilidad al nivel del sistema. Desarrollar y escalar agentes que realizan complejos razonamientos geoespaciales, incluida la interpretación, generación y raciocinio sobre mapas y configuración de datos espacial. Proyectar y mejorar sistemas de captura en grandes colecciones de documentos estáticos y semiestructurados, permitiendo que los agentes superpongan el contexto de alta señal de manera eficaz. Aprenderá a desarrollar modelos de prueba de seguridad para mejorar la precisión de misiones-datos. El rango que se muestra en cada puesto de trabajo refleja el objetivo mínimo y máximo para los nuevos salarios de contratación para la posición, determinado por la ubicación laboral y factores adicionales, incluidas las habilidades relacionadas con el empleo, experiencia, rendimiento de entrevista y educación o capacitación relevante. Los empleados a escala en roles elegibles también reciben una compensación basada en equidad, sujeta a la aprobación del Consejo de Administración. Su reclutador puede compartir más información sobre el rango salarial específico para su lugar preferido durante el proceso de contrate, y confirmar si el rol contratado será elegible para la concesión de capital. También recibirá beneficios incluyendo, pero no limitado a: Salud integral, cobertura dental y visión, prestaciones de jubilación, un estipendio de aprendizaje y desarrollo, y generoso PTO. Adicionalmente, este papel puede ser elegível para beneficios adicionales como un stipendio remunerativo. Su reclutador puede compartir más información sobre el rango salarial específico para su ubicación preferida durante el proceso de contratación, y confirmar si la función contratada será elegible para una subvención equitativa. También recibirá beneficios que incluyen, pero no se limitan a: cobertura integral de salud, odontología y visión, beneficios de jubilación, un estipendio de aprendizaje y desarrollo, y generoso PTO. Adicionalmente, este papel puede ser eligible para beneficios adicionales como un stipendio para viajeros. Trabajamos en estrecha colaboración con líderes de la industria como Meta, Cisco, DLA Piper, Mayo Clinic, Time Inc., el Gobierno de Qatar y agencias gubernamentales estadounidenses que incluyen al Ejército y la Fuerza Aérea. Estamos expandiendo nuestro equipo para acelerar el desarrollo de aplicaciones de IA. Creemos que todos deben ser capaces de llevarse a sí mismos al trabajo, por lo que estamos orgullosos de ser un lugar de trabajo inclusivo e igualitario. Estamos comprometidos con igualdad de oportunidades de empleo independientemente de raza, color, ascendencia, sexo, origen nacional, orientación sexual, edad, ciudadanía, estatus matrimonial, condición de incapacidad, género o estatus de veterano. Nos hemos comprometido a trabajar y proporcionar soluciones razonables a los solicitantes con discapacidades físicas y mentales. Si necesita asistencia y/o una adaptación razonable en el proceso de solicitud o reclutamiento debido a un interés personal, póngase en contacto con nosotros. Cualquier información que recopilemos en relación con su solicitud será tratada de acuerdo con nuestras políticas internas y programas diseñados para proteger los datos personales.