Software Engineer, ML Infrastructure

Value Driven Solutions
Costa Mesa, CA, United States
19 days ago
Apply on www.indeed.com
Prepare application

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Experience level
Expert
Compensation
$166,000.0 - $220,000.0
Working hours
Regular working hours
Job source

Tech stack

Airflow Data Infrastructure Distributed Computing Environment Python (Programming Language) Machine Learning Tensorflow Workflow Management Systems Pytorch Kubernetes Machine Learning Operations

Job description

VDS is recruiting an ML infrastructure engineer to build the training and data platform behind autonomous perception systems used across a defense-technology company.

What you’ll do

  • Build and run training and dataset infrastructure, including multi-sensor data indexing.
  • Improve tools and SDKs used by perception engineers.
  • Use frontier models for data curation, labeling, synthetic data and search.
  • Track training jobs, evaluations and model versions.
  • Partner with engineering teams in each business line.

What you bring

  • Strong modern Python; some Rust (PyO3) is a plus.
  • Distributed training in TensorFlow or PyTorch.
  • Kubernetes and workflow orchestration (Flyte, Airflow or Dagster).

Nice to have

  • Experiment tracking with Weights & Biases or MLflow.

Requirements

  • [confirm years] years of production ML engineering.
  • Python and PyTorch or TensorFlow; experience deploying models at scale.

Benefits & conditions

Type: Direct hire, full-time · Base: $166,000-$220,000 + equity · Clearance: U.S. Person required; Top Secret eligibility desirable

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on www.indeed.com
Prepare application

Good distractions

Loading talks and stories from around this role…