Software Engineer - ML Infrastructure

Anduril Anduril · Defense · Costa Mesa, CA · AFS : Discovery Engineering : ML Infrastructure

Software Engineer focused on building and operating ML infrastructure, including data management, training systems, and perception platforms for edge devices in the defense industry. The role involves improving tooling for perception engineers, integrating frontier models for data augmentation, and enabling ML lifecycle observability.

What you'd actually do

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

Skills

Required

  • Python
  • Rust
  • Tensorflow
  • PyTorch
  • distributed training
  • MLOps tooling
  • Kubernetes
  • backend services

Nice to have

  • PyO3
  • Maturin
  • Modal
  • Flyte
  • Airflow
  • Dagster
  • Weights & Biases
  • MLFlow
  • US Top Secret clearance

What the JD emphasized

  • core model training & dataset management infrastructure
  • perception engineer-facing tooling & SDKs
  • frontier models
  • ML lifecycle observability
  • edge devices

Other signals

  • ML infrastructure
  • data management systems
  • training systems
  • perception engineers
  • edge devices