Machine Learning Engineer

Axon - Seattle, WA

Full-time - Mid Level

Seattle, WA

Ambulatory Health Care Services

About the position

As a Machine Learning Engineer for Embedded AI at Axon, you will play a crucial role in developing AI solutions that enhance public safety. This position involves collaborating with a multidisciplinary team to enable new AI capabilities for Axon's products, including Fleet, Axon Body, and Axon Air. You will be responsible for the training, evaluation, testing, and deployment of machine learning models on Axon devices, contributing to the company's mission of accelerating justice and saving lives.

Responsibilities

• Collaborate with scientists and product managers to build proof-of-concepts (POCs) for future AI solutions.
• Architect and develop secure, privacy-preserving solutions for continuous improvement of AI models.
• Design the next generation of the continuous deployment framework for AI on devices.
• Optimize models for edge deployment to enable new AI features on existing and new Axon devices.
• Work with scientists to implement state-of-the-art training techniques.

Requirements

• Bachelor's Degree in Computer Science, Engineering, Electronics, Mathematics, or a related technical field.
• 6+ years of software engineering experience with a proven track record of deploying AI models on devices.
• Experience with GPU/TPU development and debugging.
• Proficiency in Python and C++, with familiarity in ML frameworks such as TensorFlow or PyTorch.
• Advanced knowledge and hands-on experience with Linux.
• Excellent problem-solving skills and ability to understand system architecture.
• Strong software design skills.
• Effective communication skills to interact with scientists, engineers, and product managers.

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
• Snacks in the offices