Staff AI Infrastructure Engineer

Anduril • Seattle, CA

2d

About The Position

We are looking for a founding Staff AI Infrastructure Engineer to architect, build, and scale the end-to-end machine learning platform that powers Anduril’s autonomous systems. As a Staff Engineer, you will own the technical roadmap for our ML platform. You will build the robust infrastructure, MLOps tooling, and systems architecture required to train, evaluate, host, and serve complex AI models (including LLMs, computer vision, and RL agents) in both cloud environments and air-gapped, offline tactical edge networks. You will be a force multiplier for our AI Research Scientists, optimizing their experimentation velocity and managing the lifecycle of terabytes of multi-modal sensor and simulation data. Over time, you will help recruit, mentor, and expand this infrastructure engineering team.

Requirements

• Production Infrastructure Experience: 7+ years of software engineering experience with a proven track record of designing, building, and operating production-scale machine learning systems and platforms (MLOps).
• Strong Programming & Systems Design: Proficient in Python, Go, C++, or similar backend languages. Deep understanding of ML systems design, memory management, and distributed computing.
• GPU & Compute Orchestration: Deep experience with containerized deployments (Docker, Kubernetes), GPU scheduling/orchestration, and distributed training frameworks (e.g., PyTorch Distributed, Ray, Slurm, or Megatron-LM).
• Data Architecture: Hands-on experience building distributed data pipelines (ETL) and managing massive datasets (terabytes of unstructured/multi-modal sensor data).
• Strategic Technical Leadership: Experience setting technical direction, leading complex system migrations, and mentoring senior engineers.
• Clearance Eligibility: Eligible to obtain and maintain an active U.S. Top Secret security clearance.

Nice To Haves

• Experience building and running ML infrastructure, model serving, or software registries within secure, air-gapped, or highly regulated environments (e.g., IL5/IL6, GovCloud).
• Experience specifically building training and evaluation platforms for Large Language Models, Generative AI architectures, or Reinforcement Learning (RL) pipelines.
• Experience profiling training hardware performance, identifying bottlenecks across networks and memory, and optimizing hardware utilization.
• Experience designing and operating multi-tenant ML platforms that serve multiple research teams, with robust resource isolation, quota management, and fair scheduling across shared GPU clusters.
• Hands-on experience with next-generation AI accelerators beyond standard GPUs (e.g., AWS Trainium, Google TPUs, or custom ASICs) for training and inference workloads.
• Experience building production monitoring and observability systems for ML models, including prediction drift detection, data quality monitoring, and automated retraining triggers.

Responsibilities

• Architect the ML Platform: Design, build, and maintain our foundational training, orchestration, and experimentation infrastructure to support state-of-the-art model development.
• Optimize AI Researcher Velocity: Actively identify, measure, and eliminate bottlenecks in the ML research lifecycle. Build highly automated tools for hyperparameter tuning, model profiling, and experimentation tracking.
• Build Data Pipelines at Scale: Design and scale robust, high-performance ETL pipelines capable of processing terabytes of multi-modal data (video, camera feeds, radar, flight telemetry, and simulation logs) captured from physical assets and test sites.
• Deploy Model Serving Infrastructure: Architect high-throughput, low-latency model serving frameworks optimized for both scalable cloud environments and air-gapped, resource-constrained tactical edge environments. Build CI/CD pipelines for ML models with automated validation, canary deployments, and rollback capabilities.
• Implement Evaluation & Alignment Loops: Build robust, automated pipelines for continuous evaluation, model validation, and reinforcement learning alignment loops (RLHF/DPO) to guarantee model safety and predictability in high-stakes environments.
• Collaborate and Pioneer: Work closely with AI Researchers, Computer Vision teams, and platform engineers to design unified infrastructure standards across the company's autonomous systems programs.

Benefits

• Highly competitive equity grants are included in the majority of full time offers; and are considered part of Anduril's total compensation package.
• 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