Software Engineer, Data Infrastructure

ABOUT THE ROLE Scale AI is seeking a highly skilled and motivated Software Engineer to join our dynamic Public Sector Engineering team. As a part of this team, you will play a critical role in supporting Scale's government customers by scoping and developing onsite solutions. Our scalable, high-performance platform is the foundation for these customer solutions, and your expertise will be instrumental in designing and implementing systems that can handle interactions with existing customer systems to help our products integrate into existing customer workflows. WHAT YOU'LL DO As a Software Engineer, Data Infrastructure, you will be responsible for architecting and building the foundational data infrastructure that will serve as the brain of a project ecosystem. You will design highly novel data models and processing pipelines capable of handling massive quantities of output data from complex simulations. At the core of this role is the challenge of building a foundational data ensemble—a unified architecture that seamlessly aggregates, structures, and stages diverse sources of simulation outputs and user inputs. Key responsibilities include: - Architecting the Data Ensemble: Design and implement the architecture to ensemble various sources of injected context (deeply structural simulation data, historical game states, and dynamic user inputs) into a unified, highly queryable format optimized for Large Language Model (LLM) consumption. - Massive Batch Infrastructure: Build highly scalable, resilient data architectures from scratch. You will optimize for moving, transforming, and processing massive quantities of simulation output data via enormous batch jobs, maintaining the minimal latency required for rapid wargame iterations. - Complex Data Modeling: Design sophisticated, highly relational data models that accurately represent massive, state-based simulation environments, making them easily interpretable by machine learning models. - First-Principles Problem Solving: Navigate highly ambiguous product requirements to design custom, ground-up systems where existing open-source or enterprise tools simply cannot handle the structural complexity or scale. - Technical Leadership: Set the technical standard for the data infrastructure team, driving rigorous code quality, system performance, and architectural clarity. WHAT YOU'LL NEED - Experience: 5+ years of backend or data infrastructure experience, operating at a Senior, Staff, or Principal level. - Engineering Excellence: Deep, expert-level proficiency in systems languages (e.g., Rust, Go, C++, or highly optimized Python/Java, Spark, PySpark) and a fundamental understanding of memory management, compute limits, and distributed systems architecture. - High-Throughput / Low-Latency Data: Proven track record of processing massive datasets. You understand how to optimize massive batch jobs and parallel processing across distributed simulation nodes without sacrificing speed. - Information Retrieval & Context Surfacing: You don't need a background in AI agents, but you must be an expert in surfacing the right needle from an ocean of hay to feed decision-making engines. We highly value engineers with backgrounds in Search & RecSys, Gaming / MMOs, and High-Frequency Trading (HFT). - Mission-Driven: A strong desire to build robust, foundational technology that supports national security and defense modernization. WHY REMOTE As a remote employee, you will have the flexibility to work from anywhere while contributing to the development of critical data infrastructure for our government customers. BENEFITS Scale AI offers competitive compensation packages, including base