Role overview

Scale is seeking a Machine Learning Engineer to join its Public Sector Engineering team, developing and deploying advanced machine learning systems for mission-critical government applications. You will work with state-of-the-art models in generative AI, computer vision, and reinforcement learning, leveraging Scale’s internal infrastructure and access to massive datasets to solve complex challenges for government partners.

Key responsibilities

Deploy production machine learning models developed internally and from the open-source community to solve real-world problems for government customers.
Maintain and improve production models through retraining, hyperparameter tuning, and architectural updates while preserving established performance standards.
Work with product and research teams to identify opportunities for machine learning-driven enhancements to existing and upcoming product lines.
Build and optimize large-scale machine learning infrastructure to automate and operationalize ML services.
Develop both generic models and fine-tuned variants using massive datasets for specific products and applications.
Act as a cross-functional advocate for machine learning techniques across engineering and product organizations.
Travel approximately 10% of the time for customer interactions and team collaboration.

Requirements

Active TS security clearance.
At least 2 years of experience building and deploying applied machine learning systems in production environments.
Extensive production experience with generative AI, agentic AI, natural language processing, deep learning, deep reinforcement learning, or computer vision.
Solid foundation in algorithms, data structures, and object-oriented programming.
Strong programming skills in Python with experience in TensorFlow or PyTorch.
Ability to learn new technologies quickly and manage multiple priorities in a fast-paced environment.

What makes this role worth considering

This position offers the opportunity to work on cutting-edge AI systems deployed to real-world government operations, including advanced products like Donovan and Thunderforge. You will engage with diverse technical challenges spanning large language models and computer vision, from developing agentic frameworks that integrate retrieval systems and production APIs to building generalizable vision foundation models supporting defense applications.