Machine Learning Engineer

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

Founded in 2023, Mach Industries is a rapidly growing defense technology company focused on developing next-generation autonomous defense platforms. At the core of our mission is the commitment to delivering scalable, decentralized defense systems that enhance the strategic capabilities of the United States and its allies. With a workforce of approximately 350 employees, we operate with startup agility and ambition.

Our vision is to redefine the future of warfare through cutting-edge manufacturing, innovation at speed, and unwavering focus on national security. We are dedicated to solving the next generation of warfare with lethal systems that deter kinetic conflict and protect global security.

The Role

Mach Industries is building an AI-forward autonomy stack for contested environments where GPS and other sensing are unavailable or unreliable. As a Machine Learning Engineer, you will own and scale the training, data, and edge-inference backbone that every vision and multi-sensor model on our product lines depends on for detection, tracking, search, navigation, targeting, and automatic target recognition. This is a broad, high-ownership role: you'll stand up the data and training infrastructure that lets the autonomy team iterate fast, generate synthetic data to cover the long tail, and get research-grade models running in real time on embedded hardware in flight. We are generalists, so you'll move fluidly between infrastructure, modeling, and deployment.

Key Responsibilities

• Own and evolve the training and data infrastructure the autonomy team builds on: ingestion from flight/sim/HITL, curation and mining, labeling/QA workflows, dataset versioning (DVC/Parquet), and reproducible dataset builds.
• Stand up and scale training/eval infrastructure: distributed multi-GPU training, experiment tracking, a model registry, and CI-based evaluation with regression gates plus automated field-data to retrain to validate to redeploy loops.
• Deploy and optimize models for real-time edge inference on Jetson-class hardware (quantization/pruning, TensorRT/ONNX Runtime); profile CPU/GPU and hit tight latency, throughput, and SWaP targets.
• Build and improve models across the portfolio as a hands-on IC: detection, segmentation, tracking, target/area search, classification/ATR, and multi-sensor fusion for EO/IR and auxiliary sensing.
• Generate and manage synthetic data at scale (simulation + domain randomization) to cover long-tail and degraded conditions and close sim-to-real gaps.
• Instrument runtime health, drift detection, and graceful degradation, and wire model-performance metrics back into the data and retraining loop.
• Live close to flight data with visualization, triage, and root-cause tooling so the team can go from field logs to insight and model updates rapidly.
• Partner with other autonomy disciplines across perception, localization, embedded, and flight-test to take capabilities from prototype to sim to HITL to flight to deployment.

Required Qualifications

• Strong generalist software engineering: Python for ML and tooling, plus production C++ on Linux; profiling, optimization, and rigorous testing discipline.
• Proven experience building ML data and training pipelines end to end: dataset construction, labeling/QA, augmentation, experiment tracking, and reproducible training.
• Hands-on training and fine-tuning in PyTorch across modern detection/segmentation/tracking architectures (CNN/Transformer).
• Edge and real-time deployment: model compression (INT8/FP16), runtime optimization (TensorRT/ONNX Runtime), and meeting latency/SWaP constraints on embedded GPU (Jetson-class) hardware.
• Data and MLOps infrastructure: SQL/Parquet, dataset/versioning tools, CI-based validation, and scalable multi-GPU training.
• BS/MS/PhD in CS/EE/Robotics or similar, or equivalent experience, with a track record shipping ML models to production or hardware. Senior candidates: deeper ownership of training/data infrastructure at scale.

Preferred Qualifications

• Synthetic data generation and simulation (e.g. Unreal/Isaac, domain randomization) and demonstrated sim-to-real transfer.
• EO/IR imagery experience and working with real flight/test data in challenging, degraded, or contested environments.
• Multi-modal perception and fusion (EO/IR + radar/LiDAR/RF) at the feature or decision level.
• Detection/tracking/search at scale; active learning and data-mining strategies for long-tail coverage.
• CUDA backends for performance debugging; ROS 2; NVIDIA Jetson deployment pipelines.
• Drift/dataset-shift monitoring, robustness and rare-event testing, long-horizon reliability metrics.
• Distributed training frameworks and cloud ML platforms (e.g. SageMaker); Docker for reproducibility; Rust for systems tooling.

To conform to U.S. Government export regulations, applicant must be a (i) U.S. citizen or national, (ii) U.S. lawful, permanent resident (aka green card holder), (iii) Refugee under 8 U.S.C. § 1157, or (iv) Asylee under 8 U.S.C. § 1158, or be eligible to obtain the required authorizations from the U.S. Department of State. Learn more about the ITAR here.

$120,000 - $160,000 USD

$165,000 - $220,000 USD

At Mach, we believe that compensation is more than just a paycheck. It's a comprehensive package designed to reward the time, skills, and expertise you bring to our team. We are committed to offering competitive compensation that is equitable, transparent, and aligned with market trends.

Key Elements of Our Compensation Philosophy:

• Competitive Salaries: We continuously research market rates to ensure our salaries are competitive with those in our industry and geographic location.
• Performance-Based Incentives: Our bonus structure is tied to both individual and company performance, creating a win-win scenario that motivates our team members to excel.
• Equity Participation: We offer stock options or equity to employees, ensuring that you share in the success and growth of the company.
• Benefit Packages: Our benefits include healthcare, dental and vision plans, retirement savings, and paid time off, because we understand that your well-being extends beyond the office.
• Professional Development: We allocate funds for continuing education, training, and career growth.
• Transparency: We are committed to openness about our compensation tiers and how decisions are made, ensuring trust and mutual respect.

By joining Mach, you’re not just accepting a job—you’re becoming a key part of a team that values your contributions and invests in your growth. Together, we aim to create a rewarding environment that sets the stage for long‑term success.

Data Privacy

To view Mach Industries candidate data privacy policy, please visit https://machindustries.com/privacy-policy.

Disclosures

This position may require access to information protected under U.S. export control laws and regulations, including the Export Administration Regulations (EAR) and the International Traffic in Arms Regulations (ITAR). Please note that any offer for employment may be conditioned on authorization to receive software or technology controlled under these U.S. export control laws and regulations without sponsorship for an export license.

Mach participates in E-Verify and will provide the federal government with your Form I‑9 information to confirm that you are authorized to work in the U.S.

The salary range for this role is an estimate based on a wide range of compensation factors, inclusive of base salary only. Actual salary offers may vary based on (but not limited to) work experience, education and training, critical skills, and business considerations. Highly competitive equity grants are included in most offers and are considered part of Mach’s total compensation package. Mach offers benefits such as health insurance, retirement plans, and opportunities for professional development.

Mach is an equal opportunity employer committed to creating a diverse and inclusive workplace. All qualified applicants will be treated with respect and receive equal consideration for employment without regard to race, color, creed, religion, sex, gender identity, sexual orientation, national origin, disability, uniform service, Veteran status, age, or any other protected characteristic per federal, state, or local law, including those with a criminal history, in a manner consistent with the requirements of applicable state and local laws. If you’d like to defend the American way of life, please reach out!