Senior Machine Learning Engineer II

Summary

Join Axon and contribute to building AI products that impact the world. As a Senior Machine Learning Engineer for Embedded AI, you will develop AI solutions for public safety, working with a team to enable new AI capabilities for Axon products. You will collaborate with scientists and product managers, architect secure and privacy-preserving solutions, and optimize models for edge devices. This role involves working with various AI domains and applications, including computer vision, speech recognition, and natural language understanding. The position is remote from Vietnam or hybrid from the Ho Chi Minh City office, reporting to the Machine Learning Manager. Axon offers competitive salary, benefits, and a chance to make a real difference.

Requirements

• 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

Responsibilities

• 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 feature on existing and new Axon’s devices by optimizing models for the edge
• Collaborate with scientists in architecting and implementing state-of-the-art training techniques

Preferred Qualifications

• Master’s Degree/PhD in Computer Science, Engineering, Electronics, Mathematics or an equivalent highly technical field
• Experience with model compression (quantization, model distillation, ...)
• Hands-on experience in solving Computer Vision or Natural Language Understanding problems in a business setting
• Familiarity with model encryption and privacy preserving AI techniques

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