Machine Learning Engineer

Job description

ABOUT MACH INDUSTRIES 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. 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! DATA PRIVACY TO VIEW MACH INDUSTRIES CANDIDATE DATA PRIVACY POLICY, PLEASE VISIT HTTPS://MACHINDUSTRIES.COM/PRIVACY-POLICY. By applying for a job at Mach, you acknowledge that you have read and understood this Recruiting Privacy Policy.