Staff ML Tech Lead — Edge CV for Autonomy

Mosaic.tech; Irvine (CA); On-site; USD 220,000 - 330,000; Full time

Benefits offered by this job

Equity grants

Job summary

Anduril Industries is seeking a Staff ML Tech Lead Manager to own the full ML stack for the Counter Intrusion team. You will design and train multi-sensor object detection models for edge devices that push the boundaries of sensor capabilities.

You will develop learning algorithms to optimize autonomous systems. Your work will cover end-to-end design, data collection and training infrastructure, and performance.

Qualifications

• MS/PhD in ML, Robotics or CS with emphasis on Computer Vision.
• BS in Computer Science, Machine Learning, Electrical Engineering, or related field.
• 6+ years of experience developing, benchmarking and optimizing ML algorithms on large-scale datasets.
• Strong Deep Learning and CV background.
• Proficiency in C++ development in a Linux environment.
• Experience with Python development and deep learning frameworks such as PyTorch, JAX and TensorFlow.
• Experience deploying models with TensorRT and ONNX.
• Optimize on-device inference and vision kernels across CPU/GPU/NPU.
• Track record of developing and deploying CV models from R&D to production.
• Experience writing and maintaining automated continuous integration tests.
• Knowledge of system profiling and tuning for latency, memory and power efficiency.
• Ability to conduct experiments, ablation studies and create highly detailed reports.
• Eligible to obtain and maintain a U.S. Secret security clearance

Responsibilities

• Propose and prototype innovative solutions to solve real-world problems, leveraging the latest state-of-the-art techniques in the field
• Develop and maintain core ML pipelines
• Train and deploy deep learning models for real-time applications
• Collaborate cross-functionally with camera, systems and labeling teams
• Curate datasets for evaluating performance and comparing performance trends over time
• Provide technical mentorship to other junior ML engineers

Skills

Deep Learning

Computer Vision

C++ development

Python development

TensorRT/ONNX

Linux

CI testing

System profiling

On-device inference

Security clearance

PyTorch/JAX/TensorFlow

Education

MS or PhD in ML/Robotics/CS

BS in CS/ML/EE

Tools

TensorRT

ONNX

Job description

Anduril Industries is seeking a Staff ML Tech Lead Manager to own the full ML stack for the Counter Intrusion team. You will design and train multi-sensor object detection models for edge devices that push the boundaries of sensor capabilities.

You will develop learning algorithms to optimize autonomous systems. Your work will cover end-to-end design, data collection and training infrastructure, and performance.