Jobs · Engineering · Washington

Senior Machine Learning Engineer II

Axon · Seattle, WA · 1 wk ago
HybridEngineering$157k–$251k/yrFull-time

About the role

We are seeking a seasoned Machine Learning Engineer to join a new team building agentic video and multimodal reasoning systems. As a senior engineer on this team you will own the systems and infrastructure that turn cutting-edge research into safe, reliable, and scalable AI capabilities — supporting our company’s mission to accelerate justice, protect truth, and save lives.

Responsibilities

  • Own end to end the systems and infrastructure that take multimodal and video models from research prototype to reliable, scalable production.
  • Design and operate data and ingest pipelines for large-scale video and multimodal corpora — storage, processing, labeling, and retrieval.
  • Build and scale training and evaluation infrastructure, and turn scientists’ evaluation methodology into automated, reproducible measurement systems.
  • Build inference and serving systems for real-time and batch multimodal workloads, optimizing latency, throughput, and cost.
  • Build the retrieval, indexing, and embedding infrastructure (vector search at scale) and the orchestration and tool-use plumbing behind agentic reasoning over video and multimodal data.
  • Ensure the platform is secure, privacy-preserving, and responsible by design.
  • Set technical direction for the team’s systems, raise the engineering bar, and mentor other engineers.

Requirements

  • Bachelor’s Degree in Computer Science, Engineering, Physics, Mathematics or an equivalent highly technical field.
  • 10+ years of software engineering experience and a proven track record of architecting, operating, and maintaining large-scale distributed platforms in production.
  • Deep experience building systems across the ML lifecycle: data pipelines, training and evaluation infrastructure, model serving and inference at scale, and production monitoring.
  • Experience with large-scale data systems handling video or other high-volume, high-dimensional multimodal data.
  • Strong software engineering fundamentals and proficiency in Python, along with familiarity with ML frameworks such as PyTorch or TensorFlow.
  • Hands-on experience operating cloud infrastructure at scale (AWS, GCP, or Azure), including distributed systems and container orchestration.
  • Excellent problem-solving skills and the ability to dive into system architecture, design, performance metrics, code, test plans, project plans, deployments, and operations.
  • Strong communication skills and comfort partnering closely with scientists, engineers, and product managers.

Qualifications

  • Master’s or PhD in Computer Science, Engineering, or an equivalent highly technical field.
  • Experience building video processing or video understanding pipelines at scale.
  • Experience with retrieval systems, vector search, and RAG infrastructure.
  • Experience serving foundation models, or building agentic systems and LLM tool-use orchestration.
  • Familiarity with responsible AI, de-identification, and privacy-preserving techniques.

Skills

  • Python programming.
  • Experience with ML frameworks like PyTorch or TensorFlow.
  • Experience with cloud infrastructure (AWS, GCP, or Azure).
  • Experience with large-scale data systems handling video or other high-volume, high-dimensional multimodal data.
  • Experience with distributed systems and container orchestration.
  • Experience with video processing or video understanding pipelines.
  • Experience with retrieval systems, vector search, and RAG infrastructure.
  • Experience with serving foundation models, or building agentic systems and LLM tool-use orchestration.
  • Experience with responsible AI, de-identification, and privacy-preserving techniques.

Benefits

  • Competitive salary and 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 offices.

Pay

$156,750 - $250,800 USD

Schedule

Hybrid from Seattle, WA or London, UK

Similar jobs