Senior Machine Learning Engineer, Public Sector

Scale AI
Washington DC2025-11-18

About the job

The goal of a Senior Machine Learning Engineer at Scale is to leverage techniques in the fields of generative AI, computer vision, reinforcement learning, and agentic AI to improve Scale's products and customer experience in production environments. Our machine learning engineers take advantage of robust internal infrastructure and unique access to massive datasets to deliver improvements to our customers. Our Public Sector Machine Learning team is focused on deploying cutting-edge models to mission-critical government systems through products like Donovan and Thunderforge. Our work spans multiple modalities, with a strong focus on both large language models and computer vision.

Responsibilities

Take state of the art models developed internally and from the community, use them in production to solve problems for our customers and taskers

Improve and maintain production models through retraining, hyperparameter tuning, and architectural updates, while preserving core performance characteristics

Collaborate with product and research teams to identify and prototype ML-driven product enhancements, including for upcoming product lines

Work with massive datasets to develop both generic models as well as fine tune models for specific products

Build scalable machine learning infrastructure to automate and optimize our ML services

Serve as a cross-functional representative and advocate for machine learning techniques across engineering and product organizations

Qualifications

Minimum

Extensive experience with GenAI, Agentic AI, natural language processing, deep learning and deep reinforcement learning, or computer vision in a production environment

Solid background in algorithms, data structures, and object-oriented programming

Strong programing skills in Python, experience in Tensorflow or PyTorch

Preferred

Graduate degree in Computer Science, Machine Learning or Artificial Intelligence specialization

Experience working with cloud platforms (eg. AWS or GCP) and deploying machine learning models in cloud environments

Experience with computer vision, generative AI models, large language models, or agentic systems

Familiarity with ML evaluation frameworks and agentic model design