Software Engineer - ML Infrastructure

PVH (Tommy Hilfiger/Calvin Klein); Costa Mesa (CA); On-site; USD 191,000 - 253,000; Full time

Benefits offered by this job

Equity grants

Comprehensive benefits

Job summary

Anduril Industries is seeking a Software Engineer for the ML Infrastructure team to build and operate core data management and model training systems for perception and autonomy use-cases. You’ll index multi-modal sensor data, orchestrate training across cloud and on-premise compute, and support deployment to edge devices.

You’ll contribute to frontier model integration, develop perception datasets, and enable observability across the ML lifecycle while partnering with perception engineers and

Qualifications

• Build and operate core training & dataset management infrastructure for perception engineers.
• Improve perception engineer-facing tooling & SDKs and iterate with users.
• Integrate frontier models for data augmentation, curation, and synthetic data generation.
• Enable observability & traceability across the ML lifecycle.
• Eagerly engage with customers to understand data requirements.

Responsibilities

• Develop core data management systems and model training orchestration across cloud/on-premise compute.
• Index and manage multi-modal sensor data for perception tasks.
• Train, evaluate, and deploy models to edge devices.
• Create tooling for monitoring training jobs and storing evaluation reports.
• Collaborate with customers and cross-functional teams to deliver solutions.

Skills

ML infrastructure

SDKs & tooling

Data augmentation

Observability

Customer engagement

Tools

Python

Rust

TensorFlow

PyTorch

Kubernetes

MLOps tooling

Dagster

Flyte

Airflow

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:

Benefits

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 .

Protect