Axon is hiring an Applied Machine Learning Engineer to shape the mathematical foundations of our real-time airspace awareness systems. In this role, you will research, design, and validate advanced estimation and probabilistic models for multi-sensor fusion and multi-target tracking.
What You'll Do
- Research and develop advanced multi-sensor fusion and multi-target tracking methodologies.
- Design probabilistic models and state estimation frameworks for radar, RF, optical, and other sensing modalities.
- Develop and evaluate algorithms such as Kalman filters (EKF/UKF), particle filters, Bayesian filters, probabilistic data association, and multi-hypothesis tracking.
- Conduct simulation studies and performance benchmarking across varying operational conditions and noise environments.
- Analyze real-world datasets to validate model assumptions, quantify uncertainty, and improve robustness.
- Partner closely with software engineers to transition validated algorithms into scalable, production-ready systems.
- Publish internal technical documentation and contribute to intellectual property development where applicable.
- Stay current with advancements in tracking, estimation theory, probabilistic modeling, and machine learning research.
What We're Looking For
- Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, Applied Mathematics, or related field — or equivalent hands-on experience building ML systems in production.
- 8+ years of experience developing and deploying machine learning systems in real-world applications.
- Well-developed proficiency in Python and experience with ML frameworks such as PyTorch, TensorFlow, or similar.
- Experience deploying ML systems in production environments (cloud, edge, or embedded systems).
- Solid understanding of statistics, model evaluation, and performance trade-offs.
- Familiarity with MLOps practices including CI/CD for ML, model versioning, monitoring, and automated retraining workflows.
- Experience working with large, noisy, or multi-modal datasets.
- Clear communication skills and a collaborative mindset aligned with Axon’s values.
Technical Stack
- Python
- PyTorch
- TensorFlow
Benefits & Compensation
- Competitive salary: $150,750 — $241,200 USD
- 401k with employer match
- Discretionary paid time off
- Paid parental leave for all
- Medical, Dental, Vision plans
- Fitness Programs
- Emotional & Mental Wellness support
- Learning & Development programs
- Employee Resource Groups (ERGs)
- Snacks in our offices
Work Mode
This is a hybrid position based in Seattle, WA.
We are an equal opportunity employer that promotes justice, advances equity, values diversity and fosters inclusion.
