Machine Learning Engineer II at Axon

Axon is seeking a Machine Learning Engineer II to join their team in Vietnam. The Machine Learning Engineer II will work closely with other ML engineers and scientists, enabling new AI capabilities for Axon’s products by supporting the training, evaluation, testing, and deployment of machine learning models on Axon devices and cloud. The person will contribute to building AI products impacting the world for the greater good and developing AI solutions transforming the public safety space.

Role Involves:

• Collaborating with scientists and product managers to build proof-of-concepts.
• Architecting and developing secure, privacy-preserving solutions.
• Architecting the next generation of the continuous deployment framework for AI on device.
• Enabling new AI feature on existing and new Axon’s devices by optimizing models for the edge.
• Collaborating with scientists in architecting and implementing state-of-the-art training techniques.

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 or cloud environments (AWS, Azure).
• 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.

What Axon Offers:

• An awesome office/working environment with the latest technologies
• Competitive salary and benefits including a great medical insurance plan
• Fun team activities and outings
• Flexible working hours