Senior Machine Learning Engineer II

Axon Axon · Enterprise · Office, WA · 2014 Artificial Intelligence

Senior Machine Learning Engineer II at Axon focused on architecting and implementing on-device AI solutions for edge devices. This role involves optimizing models for limited memory and connectivity, enabling continuous improvement, and implementing model compression techniques. Collaboration with scientists and device engineers is key to bringing AI capabilities to Axon products.

What you'd actually do

  1. Architect and develop secure, privacy-preserving, on device solutions to enable the continuous improvement of existing AI models.
  2. Collaborate with scientists in architecting and implementing state-of-the-art edge distributed training techniques.
  3. Implement on device monitoring solutions used for continuous model improvement
  4. Implement innovative model compression solutions to enable AI at the edge.
  5. Impact the team by bringing your own expertise and deep knowledge of the state-of-the-art to introduce new techniques leading to tangible impact in terms of model fairness, performance, and platform scalability.

Skills

Required

  • Python
  • C++
  • TensorFlow
  • PyTorch
  • on chips development
  • system architecture
  • design
  • performance metrics
  • code
  • test plans
  • project plans
  • deployments
  • operations

Nice to have

  • Computer Vision
  • Natural Language Understanding
  • responsible AI
  • de-biasing
  • model encryption
  • de-identification techniques

What the JD emphasized

  • successfully optimizing models for the edge
  • deploying AI models at the edge
  • on device with limited memory & connectivity
  • continuous model improvement
  • model compression solutions

Other signals

  • deploying AI models at the edge
  • on device with limited memory & connectivity
  • continuous model improvement
  • model compression solutions