Software Engineer - ML Infrastructure

Anduril Industries, Inc.; Costa Mesa (CA); On-site; USD 191,000 - 253,000; Full time

Job summary

Anduril Industries in Costa Mesa, CA, is hiring a Software Engineer on the ML Infrastructure team to build core data management, model training orchestration across cloud and on-prem, and the internal perception platform.

You will enable dataset querying, data curation, and model deployment to edge devices, working with frontier models, orchestration frameworks, and observability tools to support fast, reliable ML pipelines.

Qualifications

• Experience building model training and dataset management infrastructure.
• Ability to partner with perception engineers to improve tooling.
• Experience with MLOps tooling and ML lifecycle observability.

Responsibilities

• Build and operate core training and dataset management infrastructure.
• Improve perception engineer-facing tooling and SDKs.
• Integrate frontier models for data augmentation, curation, and synthetic data generation.
• Enable observability and traceability of the ML lifecycle.
• Engage with customers to understand data requirements and constraints.

Skills

Python

Rust

Machine Learning

MLOps

TensorFlow

PyTorch

Kubernetes

Cloud & On-Prem

Model Training

Tools

PyO3

Maturin

Modal

Flyte

Airflow

Dagster

Weights & Biases

MLFlow

Job 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.
• 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 our model registry. This integrates with our Fleet Management systems to continuously deliver model improvements to our edge systems.
• 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 needs and constraints.

PREFERRED QUALIFICATIONS

• 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.

US Salary Range

$191,000 - $253,000 USD

The salary range for this role is an estimate based on a wide range of compensation factors, inclusive of base salary only. Actual salary offer may vary based on (but not limited to) work experience, education and/or training, critical skills, and/or business considerations. Highly competitive equity grants are included in the majority of full time offers; and are considered part of Anduril's total compensation package. Additionally, Anduril offers top-tier benefits for full-time employees, including:

At Anduril, we invest in our people. Our comprehensive, competitive benefits package (available at little to no cost to employees) ensures you’re supported in health, recovery, and whatever comes next. For more information, Explore Our Benefits .