A

Axon

Builds weapons, cameras and software for police

Senior Machine Learning Engineer

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeSeattle, WA, USA

About the Job

Requirements

  • A bachelor’s degree in Computer Science, Engineering, Electronics, Mathematics, or another highly technical field is required.
  • At least 6 years of software engineering experience, including a proven track record of deploying AI/ML models to production in the cloud.
  • Strong experience with cloud architecture and infrastructure as code.
  • Proficiency in Python and C++, with experience using machine-learning frameworks such as PyTorch or TensorFlow.
  • Advanced, hands-on knowledge of Linux environments and systems.
  • Strong problem-solving skills and curiosity to investigate complex system architectures.
  • Software architecture and design skills focused on building robust, scalable, maintainable systems.
  • Ability to communicate and collaborate across disciplines with scientists, engineers, and product managers.

Responsibilities

  • Partner with scientists, engineers, and product managers to rapidly prototype emerging AI research, build proof-of-concepts, and help determine which technologies can become future Axon AI products.
  • Architect and develop infrastructure to train, evaluate, deploy, monitor, and continuously improve models across Axon’s cloud and device ecosystem.
  • Address engineering challenges involving model training and fine-tuning, large-scale evaluation, inference optimization, data pipelines, cloud infrastructure, and production machine-learning systems.
  • Design reusable platforms, infrastructure, and developer tooling to help scientists and engineers experiment, evaluate ideas, and move models from research into production.
  • Explore techniques for evaluating, fine-tuning, and optimizing large models, including model distillation, quantization, and scalable evaluation.
  • Help connect cloud-based AI with models operating in resource-constrained environments while balancing model quality, latency, reliability, privacy, and cost.
  • Architect solutions that allow models to improve continuously while maintaining rigorous security, privacy, and responsible data-use standards.
  • Translate state-of-the-art research into robust engineering systems and help develop training, evaluation, and deployment techniques applicable across Axon products.
  • Own technical problems end to end, influence architecture, make key technical decisions, and advance work from early experiments to production products.
  • Help establish best practices for machine-learning engineering, architecture, testing, observability, reproducibility, and responsible AI.
  • Work across computer vision, speech, natural language processing, generative AI, edge machine learning, and cloud machine-learning infrastructure.
  • Develop expertise in systems problems at the intersection of machine learning, distributed systems, cloud infrastructure, and production software engineering.
  • Progress from owning individual systems to shaping architectures, technical roadmaps, and engineering practices across teams and products.
  • Work with machine-learning scientists, product teams, and engineers to connect advanced research with production AI systems.
  • As technical scope grows, mentor engineers, lead technical initiatives, influence cross-team decisions, and help define how AI is built at Axon.

Desired Qualifications

  • A master’s degree or PhD in Computer Science, Engineering, Electronics, Mathematics, or another highly technical field.
  • Experience with LLMOps, including model evaluation, monitoring, quantization, distillation or teacher-student approaches, and production deployment.
  • Hands-on experience fine-tuning and optimizing large-scale models.
  • Familiarity with model encryption, privacy-preserving machine learning, and secure AI techniques.

About the Company

Axon

Builds weapons, cameras and software for police

Axon builds public safety technology for police, emergency responders and security teams, combining connected devices with software that manages their work. Its TASER weapons and body cameras link field equipment to a cloud platform used throughout an investigation. Agencies use Axon Evidence to store recordings, control access and share case material with prosecutors, while its records and real-time operations tools organize reports and coordinate responses. Axon earns money from device sales and subscriptions that combine software, equipment upgrades and support over multiple years. Founded in 1993, Axon is headquartered in Scottsdale, Arizona.

Company size
5,001-10,000
Company stage
IPO
Headquarters
Scottsdale, AZ
Founded
1993

Recently Posted Jobs

Explore more opportunities at Axon

See more jobs