Axon

Sr Machine Learning Engineer II

Axon
Apply
5 hours ago

Base Salary

$151k - $241k/yr

Responsibilities

  • 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 Kalman filters, particle filters, Bayesian filters, probabilistic data association, and multi-hypothesis tracking algorithms.
  • Conduct simulations and performance benchmarking across operational conditions and noise environments.
  • Analyze real-world datasets to validate model assumptions, quantify uncertainty, and improve robustness.
  • Partner with software engineers to transition validated algorithms into scalable, production-ready systems.
  • Document technical work and contribute to intellectual property development where applicable.
  • Stay current with tracking, estimation theory, probabilistic modeling, and machine learning research.

Requirements

  • Bachelor’s or master’s degree in computer science, electrical engineering, applied mathematics, or a 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.
  • Proficiency in Python and experience with ML frameworks such as PyTorch, TensorFlow, or similar.
  • Experience deploying ML systems in cloud, edge, or embedded production environments.
  • Strong understanding of statistics, model evaluation, and performance trade-offs.
  • Familiarity with MLOps practices including model versioning, monitoring, and automated retraining workflows.
  • Experience working with large, noisy, or multimodal datasets.
  • Clear communication and collaborative working skills.

Benefits

  • Competitive salary, 401k with employer match, discretionary paid time off, paid parental leave, medical, dental, and vision plans.
  • Fitness programs, emotional and mental wellness support, learning and development programs, employee resource groups, and office snacks.
  • Hybrid work in Seattle, with onsite work Tuesday through Friday and remote flexibility on Mondays unless an accommodation is approved.

Tech Stack

PythonPyTorchTensorFlow

Categories

Axon

About Axon

5,001-10,000 employees

Axon builds TASER energy devices, body and in-car cameras, sensors, and a cloud software suite for managing evidence, records, and real-time operations used by law enforcement, public safety agencies, and enterprises. Revenue comes from hardware sales bundled with SaaS subscriptions and services across its Axon Evidence and Records platforms. Founded in 1993 and headquartered in Scottsdale, Arizona, Axon Enterprise is a NASDAQ-listed public company serving police departments and government customers globally.

Contact me