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 a month 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

RF Machine Learning Engineer

Senior RF Machine Learning Engineer position at Axon, combining DSP and ML expertise to develop innovative wireless communication solutions in a hybrid work environment.

Machine Learning Engineer

Senior Machine Learning Engineer role at Axon focusing on developing advanced airspace security systems using computer vision and ML algorithms.

Sr. Research Scientist

Senior Research Scientist position at Axon, focusing on AI/ML research and development for public safety applications, offering competitive compensation and remote work options.

Machine Learning Engineer

Senior Machine Learning Engineer role at Axon focusing on developing advanced airspace security systems using computer vision and ML algorithms.

ML Engineer for Embedded AI

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