Machine Learning Engineer, Public Sector

The goal of a 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. On the LLM side, we are developing agentic systems that help solve complex operational and planning challenges for government partners. This includes building agent frameworks that integrate with custom retrieval pipelines and production APIs, as well as evaluation tools to benchmark and refine agent behavior. We're also advancing research in areas like reinforcement learning for agentic LLMs, with successful deployment into real-world operational environments. On the computer vision front, we're training advanced models to increase labeling throughput and automate perception tasks. Our efforts include building large-scale fine-tuning pipelines, training models across multiple modalities, and developing generalizable vision foundation models to support a wide range of defense applications.

You will:

• 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
• Be comfortable learning new technologies quickly and managing multiple priorities in a fast-paced environment
• This role will require an active security clearance or the ability to obtain a security clearance.

Ideally You’d Have:

• Extensive experience using computer vision, deep learning and deep reinforcement Learning, or natural language processing in a production environment
• Solid background in algorithms, data structures, and object-oriented programming
• Strong programing skills in Python, experience in Tensorflow or PyTorch

Nice to Haves:

• 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

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, we believe that the transition from traditional software to AI is one of the most important shifts of our time. Our mission is to make that happen faster across every industry, and our team is transforming how organizations build and deploy AI.  Our products power the world's most advanced LLMs, generative models, and computer vision models. We are trusted by generative AI companies such as OpenAI, Meta, and Microsoft, government agencies like the U.S. Army and U.S. Air Force, and enterprises including GM and Accenture. 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.