Software Engineer - ML Infrastructure

Anduril • Costa Mesa, CA

3d

About The Position

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

Requirements

• U.S. Person status is required as this position needs to access export controlled data.

Nice To Haves

• 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
• Eligibility to obtain/maintain a US Top Secret clearance is also desirable.

Responsibilities

• Build & operate our core model training & dataset management infrastructure
• Improve perception engineer-facing tooling & SDKs
• Integrate frontier models for data augmentation, curation, and synthetic data generation
• Enable observability & traceability of the full ML lifecycle
• Eagerly engage with our customers

Benefits

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

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed