Axon

Senior Machine Learning Engineer I

Axon
Apply
5 hours ago

Base Salary

$151k - $241k/yr

Responsibilities

  • Own the systems and infrastructure that move multimodal and video models from research prototypes to reliable, scalable production.
  • Design and operate large-scale video and multimodal data ingestion, storage, processing, labeling, and retrieval pipelines.
  • Build training, evaluation, inference, and serving infrastructure for real-time and batch workloads.
  • Develop retrieval, indexing, embedding, vector search, orchestration, and tool-use infrastructure for agentic reasoning.
  • Ensure the platform is secure, privacy-preserving, reliable, and responsible by design.
  • Set technical direction, raise engineering standards, and mentor other engineers.

Requirements

  • Bachelor’s degree in Computer Science, Engineering, Physics, Mathematics, or an equivalent highly technical field.
  • 7+ years of software engineering experience and a track record architecting, operating, and maintaining large-scale distributed production platforms.
  • Deep experience across the ML lifecycle, including data pipelines, training and evaluation infrastructure, model serving, inference, and production monitoring.
  • Experience with high-volume video or multimodal data systems and distributed cloud infrastructure.
  • Strong software engineering fundamentals and proficiency in Python, with familiarity with PyTorch or TensorFlow.
  • Experience with container orchestration and cloud platforms such as AWS, GCP, or Azure.
  • Strong problem-solving, communication, architecture, performance, deployment, and operations skills.
  • Preferred qualifications include a master’s or PhD, video processing or understanding experience, retrieval and vector search experience, RAG infrastructure experience, foundation model serving, agentic systems, LLM tool-use orchestration, responsible AI, de-identification, or privacy-preserving techniques.

Benefits

  • Competitive salary and 401k with employer match.
  • Discretionary paid time off and paid parental leave for all.
  • Medical, dental, and vision plans.
  • Fitness programs and emotional and mental wellness support.
  • Learning and development programs and employee resource groups.
  • Seattle, WA hybrid schedule with onsite work Tuesday through Friday and remote flexibility on Mondays.
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