ML Engineer - Edge AI for Autonomous Defense Systems

Mach Industries; San Francisco (CA); On-site; USD 120,000 - 220,000; Full time

Benefits offered by this job

Equity participation

Healthcare

Professional development

Paid time off

Job summary

Mach Industries is seeking a Machine Learning Engineer to own and scale the training, data, and edge-inference backbone across vision and multi-sensor models. You will build data infrastructure, manage synthetic data, and run research-grade models in real time on embedded hardware in flight.

The role requires deep experience with Python/C++, PyTorch, and deployment on Jetson-class platforms, with a focus on low latency and SWaP constraints. A BS/MS/PhD in CS/EE/Robotics is preferred.

Qualifications

• Strong generalist software engineering with Python and C++ on Linux.
• Proven experience building ML data and training pipelines end to end.
• Hands-on training and fine-tuning in PyTorch across detection/segmentation/tracking.

Responsibilities

• Own and evolve training/data infrastructure for the autonomy team.
• Scale training/eval infrastructure with multi-GPU setups and model registry.
• Deploy and optimize real-time edge models on Jetson-class hardware.

Skills

Python

C++

PyTorch

ML Ops

Linux

Profiling

Education

BS/MS/PhD in CS/EE/Robotics

Tools

Jetson

TensorRT

ONNX Runtime

Parquet

DVC

CUDA

Job description

Mach Industries is seeking a Machine Learning Engineer to own and scale the training, data, and edge-inference backbone across vision and multi-sensor models. You will build data infrastructure, manage synthetic data, and run research-grade models in real time on embedded hardware in flight.

The role requires deep experience with Python/C++, PyTorch, and deployment on Jetson-class platforms, with a focus on low latency and SWaP constraints. A BS/MS/PhD in CS/EE/Robotics is preferred.