Clash
of Jobs
Browse All Jobs
Job Description
Axon is seeking a Senior Machine Learning Engineer II to contribute to building AI products that impact public safety. This role involves developing AI solutions for Axon's products, supporting the training, evaluation, testing, and deployment of machine learning models on Axon devices. The engineer will collaborate with scientists and product managers to build proof-of-concepts and architect secure, privacy-preserving solutions.
What This 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 features 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.
- 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:
- Competitive salary
- 401k with employer match
- Discretionary paid time off
- Paid parental leave
- Medical, Dental, Vision plans
- Fitness Programs
- Emotional & Mental Wellness support
- Learning & Development programs
