ML Engineer for Embedded AI

Axon develops devices and cloud software for public safety, focusing on protecting life through technology and innovative solutions.
$108,750 - $174,000
Machine Learning
Senior Software Engineer
In-Person
6+ years of experience
AI · Enterprise SaaS

Description For ML Engineer for Embedded AI

Axon is seeking a Machine Learning Engineer for Embedded AI to join their mission of protecting life through innovative technology. This role focuses on developing AI solutions that transform public safety, working across various domains including computer vision, speech recognition, and natural language understanding. The position offers an opportunity to work on impactful projects, optimizing and deploying AI models on various Axon devices including Fleet, Axon Body, and VR systems.

The role combines technical expertise with meaningful impact, requiring strong skills in Python, C++, and Linux, along with experience in AI model deployment on devices. You'll work in a collaborative environment, partnering with scientists and product managers to develop cutting-edge solutions while maintaining privacy and security standards.

Axon offers a comprehensive benefits package including competitive salary (108,750 - 174,000 USD), 401k matching, healthcare benefits, and various wellness programs. The company promotes an inclusive culture and welcomes diverse perspectives, encouraging applications even if candidates don't meet every qualification.

This position is available in several hub locations including Seattle and Boston, offering the opportunity to work on transformative technology while contributing to Axon's mission of protecting life and pursuing critical safety and justice issues. The role provides significant growth opportunities in a fast-paced, meaningful environment where innovation and impact go hand in hand.

Last updated 5 days ago

Responsibilities For ML Engineer for Embedded AI

  • Collaborate with scientists and product managers to build proof-of-concepts (POCs)
  • Architect and develop secure, privacy-preserving solutions for AI model improvement
  • Architect the next generation of continuous deployment framework for AI on device
  • Enable new AI features by optimizing models for the edge
  • Collaborate with scientists in architecting and implementing state-of-the-art training techniques

Requirements For ML Engineer for Embedded AI

Python
Linux
  • Bachelor's Degree in Computer Science, Engineering, Electronics, Mathematics or equivalent technical field
  • 6+ years of software engineering experience with proven track record of deploying AI models on devices
  • Experience with GPU/TPU development and debugging
  • Proficiency in Python and C++, familiarity with ML frameworks like TensorFlow or PyTorch
  • Advanced knowledge and hands-on experience with Linux
  • Excellent problem solving skills and ability to dive deep into system architecture
  • Excellent software design skills
  • Strong communication skills

Benefits For ML Engineer for Embedded AI

401k
Dental Insurance
Medical Insurance
Mental Health Assistance
Parental Leave
Vision Insurance
  • 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
  • Office snacks

Interested in this job?

Jobs Related To Axon ML Engineer for Embedded AI

ML Engineer for Embedded AI

ML Engineer for Embedded AI at Axon: Develop AI solutions for public safety, optimizing models for edge devices.

Sr. Research Scientist

Sr. Research Scientist at Axon: Drive AI/ML innovation in public safety, from prototyping to deployment, with competitive pay and benefits.

Sr. Research Scientist

Sr. Research Scientist at Axon: Drive AI/ML innovation in public safety, from prototyping to deployment, with competitive pay and benefits.

Sr AI Research Scientist I - Machine Learning & MLLM's

Senior AI Research Scientist role at Axon, focusing on machine learning and computer vision for public safety applications.

Sr AI Research Scientist I - Machine Learning & MLLM's

Senior AI Research Scientist role at Axon, focusing on machine learning and computer vision for public safety applications.