Machine Learning Engineer, Senior
150k – 250kAustin, TXML EngineeringOnsite
Summary
Designs, trains, and deploys ML models for aerial target detection, classification, and tracking in a counter-drone defense system. Owns dataset pipelines, training infrastructure, and production deployment under strict edge constraints. Requires PyTorch/JAX fluency and production ML experience.
About the role
Essential Duties
- Design, train, and iterate on machine learning models for detection, classification, and tracking of aerial targets.
- Own the dataset pipeline end-to-end, including data collection, labeling, curation, augmentation, synthetic data generation, and closed-loop retraining based on field performance.
- Build and maintain training infrastructure, including experiment tracking, compute orchestration, and evaluation harnesses.
- Define metrics and evaluation methodologies that correlate to real-world operational performance.
- Support deployment of trained models into the production perception stack, and address discrepancies between training and deployed performance.
- Analyze field data to identify and address model failure modes.
Requirements
- Demonstrated experience shipping machine learning systems into production under real-world operational requirements.
- Fluency in PyTorch or JAX, including full training loop development beyond fine-tuning off-the-shelf models.
- Strong software engineering skills beyond model development, including ownership of training infrastructure.
- Proficiency in Python; working knowledge of C++ or Rust at the training-to-deployment boundary.
- U.S. citizenship and ability to pass a background check.
Preferred Qualifications
- Experience with detection or tracking of small, fast, or adversarially perturbed targets.
- Synthetic data generation and sim-to-real methodologies.
- Experience training models for edge deployment, including quantization-aware training and knowledge distillation.
- Prior experience in defense or other safety-critical machine learning applications.
- Active security clearance, or eligibility to obtain one.
- Passion for building robots or engineering projects as a hobby.
Skills
PyTorchJAXPythonC++Rustsynthetic data generationquantization-aware trainingknowledge distillationdataset pipelineexperiment tracking
Similar roles at this salary range
All ML Engineering jobs →Senior Machine Learning Operations Engineer
Build and operate Mercury's real-time ML inference platform for fraud risk decisioning. Own model deployment, observability, and lifecycle tooling with strong backend Python fundamentals.
167k – 208kSan Francisco, CA +2ML EngineeringHybrid5+ YOESQLSHAP
AI Engineer, Evaluation
Design and implement evaluation frameworks and pipelines for AI systems using Evaluation-Driven Development. Build Python-based test suites, LLM graders, and measurement systems that guide prompt iteration and production deployment decisions.
150k – 250kSan Francisco, CA +1ML EngineeringHybrid2+ YOEPythonAI Systems