CareerPlan

Senior Machine Learning Engineer

Washington-OfficeFull-time2026-08-26 → 2026-11-10
Hybrid · United States

Core

Build AI solutions transforming public safety by training, evaluating, testing, and deploying machine learning models across Axon devices and the cloud.

Role type

Senior Machine Learning Engineer (AI Infrastructure & Production)

Builds

AI systems for public safety products (Fleet, Draft One, Axon Air, VR) and reusable ML platforms

Domain

Public Safety / AI / Cloud Infrastructure

Deliverable

production ML models

Required skills

Python, C++, PyTorch, TensorFlow, Linux, Cloud Architecture, Infrastructure as Code, Model Training, Inference Optimization, Data Pipelines, System Architecture

Preferred skills

LLMOps, Model Fine-tuning, Model Quantization, Model Distillation, Privacy-preserving ML, Large-scale Model Optimization

Technologies

PyTorch, TensorFlow, Linux, Cloud Infrastructure

Responsibilities

Architect and develop infrastructure for training, evaluation, deployment, and monitoring of ML models; Build reusable platforms and developer tooling for scientists and engineers; Solve edge-and-cloud ML problems balancing quality, latency, and privacy; Collaborate with ML scientists to translate research into robust engineering systems; Establish best practices for ML engineering, testing, and responsible AI.

Seniority

Senior, hands-on IC with leadership potential