Machine Learning Engineer - Axon : Job Details

Are you eager to contribute to building AI products impacting the world for the greater good?As a ML Engineer for Embedded AI at Axon, you will contribute to developing AI solutions transforming the public safety space and supporting our company's mission: accelerate justice, protect truth, and save lives. Working closely with other ML engineers and scientists, you will enable new AI capabilities for Axon's products (Fleet, Axon Body, Axon Air, VR and more) by supporting the training, evaluation, testing and deployment of machine learning models on Axon devices. As part of a multidisciplinary team, you will be exposed to a wide range of AI domains and applications: from computer vision to speech recognition and natural language understanding.

Location: Any of our Hub Locations in Seattle, Phoenix, Atlanta or BostonReports to: Sr Manager, Machine Learning Engineering

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

• Bachelor's Degree in Computer Science, Engineering, Electronics, Mathematics or an equivalent highly technical field.
• 6+ years of software engineering experience and a proven track record of successfully deploying AI models on devices.
• Experience with GPU/TPU development and debugging.
• Proficiency in Python and C++, familiarity with ML frameworks such as 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.
• Comfort communicating and interacting with scientists, engineers, and product managers.

• 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.
• And yes, we have snacks in our offices.

Benefits listed herein may vary depending on the nature of your employment and the location where you work.

The Pay: Axon is a total compensation company, meaning compensation is made up of base pay, bonus, and stock awards. The starting base pay for this role is between USD 108,750 in the lowest geographic market and USD 174,000 in the highest geographic market. The actual base pay is dependent upon many factors, such as: level, function, training, transferable skills, work experience, business needs, geographic market, and often a combination of all these factors. Our benefits offer an array of options to help support you physically, financially, and emotionally through the big milestones and in your everyday life. To see more details on our benefits offerings please visit www.axon.com/careers/benefits.