Skip to content

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 →
Mem0

Senior Research Engineer

Own the end-to-end lifecycle of memory features for AI agents. Fine-tune models, implement research, build evaluations, and ship production systems with Engineering.

175k – 250kSan Francisco, CAML EngineeringOn-site7+ YOERAGvLLM
Mozilla

Senior Machine Learning Engineer

Senior ML Engineer focused on fine-tuning and deploying LLMs and generative AI features into Firefox, emphasizing privacy, latency, and user experience.

139k – 218kUnited StatesML EngineeringRemote4+ YOERayLangChain
Ironclad

Senior Software Engineer, AI

Lead design and delivery of high-priority AI initiatives across multiple codebases. Build and ship AI-powered features with strong backend fundamentals and product sense.

180k – 220kSan Francisco, CAML EngineeringHybrid5+ YOEReactEvals
Mercury

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
Distyl AI

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