Software Engineer - ML Infrastructure
Build and maintain ML infrastructure for autonomous military systems, including data management, model training/orchestration, and edge deployment pipelines using Python, PyTorch, and Kubernetes.
- Compensation
- $191K–$253K / yr
- Location
- Costa Mesa, California, United States
- Work setup
- Not specified
- Seniority
- Not specified
- Employment
- Full time
- Experience
- Not specified
- Category
- Software engineering
- Posted
Skills
Must have
- Build & operate our core model training & dataset management infrastructure: Perception engineers rely on reliable, high quality data to continually retrain and improve our detection & tracing systems
- Improve perception engineer-facing tooling & SDKs: Perception engineers across Anduril rely on our core model training and evaluation framework. You’ll partner closely with these engineers to understa
- Integrate frontier models for data augmentation, curation, and synthetic data generation: While this team is not ultimately responsible for delivering perception systems for particular products, you w
- Enable observability & traceability of the full ML lifecycle: You’ll build and develop systems that enable real-time monitoring of training jobs, tools to store/curate model evaluation reports, and ou
- Eagerly engage with our customers: We partner closely with business line engineering teams to understand their specific data requirements and iteratively deliver solutions that meet their particular n
Nice to have
- Strong proficiency in modern Python & exposure to Rust/the Rust ↔ Python toolchain (PyO3, Maturin, etc).
- Experience with modern deep learning frameworks such as Tensorflow or PyTorch, as well as distributed training paradigms
- Experience leveraging modern MLOps tooling on the orchestration side (Modal, Flyte, Airflow, Dagster, etc.), as well as common ML observability platforms (Weights & Biases, MLFlow, etc)
- Familiarity with building and deploying backend services into a Kubernetes environment
- U.S. Person status is required as this position needs to access export controlled data. Eligibility to obtain/maintain a US Top Secret clearance is also desirable.
Full description
Anduril Industries is a defense technology company with a mission to transform U.S. and allied military capabilities with advanced technology. By bringing the expertise, technology, and business model of the 21st century’s most innovative companies to the defense industry, Anduril is changing how military systems are designed, built and sold. Anduril’s family of systems is powered by Lattice OS, an AI-powered operating system that turns thousands of data streams into a realtime, 3D command and control center. As the world enters an era of strategic competition, Anduril is committed to bringing cutting-edge autonomy, AI, computer vision, sensor fusion, and networking technology to the military in months, not years.
ABOUT THE TEAM
Anduril’s core Lattice platform integrates heterogeneous sensors across large fleets of assets to build a single common operating picture, providing mission-critical situational awareness and context to our users. As a Software Engineer on the ML Infrastructure team, you’ll be responsible for developing the core underlying systems and infrastructure that perception engineers across the business leverage to field sophisticated, autonomous edge systems.
ABOUT THE JOB
In particular, you’ll develop our core data management systems for perception and autonomy use-cases, including indexing multi-modal sensor data, our model training/orchestration systems (that run across cloud-based and on-premise compute), as well as our internal perception platform. This platform enables dataset querying and discovery, orchestrating an internal suite of frontier models to tag and curate data, and ultimately train & evaluate models and deploy them back to edge devices. This team also owns our core training libraries/SDKs for perception engineers, and partners with them to integrate novel model architectures into our standard infrastructure so the entire company can benefit from each improvement.
REQUIRED QUALIFICATIONS
- Build & operate our core model training & dataset management infrastructure: Perception engineers rely on reliable, high quality data to continually retrain and improve our detection & tracing systems. You’ll own and operate our core training systems, as well as our dataset management infrastructure.
- Improve perception engineer-facing tooling & SDKs: Perception engineers across Anduril rely on our core model training and evaluation framework. You’ll partner closely with these engineers to understand their pain-points & feature requests, and iterate with them directly to deliver them.
- Integrate frontier models for data augmentation, curation, and synthetic data generation: While this team is not ultimately responsible for delivering perception systems for particular products, you will be responsible for integrating frontier models at all stages of our dataset curation/annotation pipeline, as well as enabling semantic search across our broad data holdings.