Senior Manager, Machine Learning Engineering

At Axon, we're on a mission to Protect Life. We're explorers, pursuing society's most critical safety and justice issues with our ecosystem of devices and cloud software. As a ML Engineer Manager at Axon, you will contribute to architecting and implementing the platform used by Axon scientists to transform the public safety space. You'll collaborate closely with scientists and engineers to enable new AI capabilities for Axon products (Fleet, Axon Body, Axon Air, and more) by supporting the training, evaluation, testing and deployment of machine learning models used across the board. You'll be exposed to a wide range of domains and applications, from computer vision to speech recognition and NLU. The ideal candidate will have a consistent track record of successfully leveraging and managing large-scale distributed platforms for secure and compliant research in the cloud, bring outstanding hands-on management skills, understand the AI lifecycle, and be capable of supporting scientists along the development of state-of-the-art AI solutions.

Key Responsibilities: • Architect, develop and manage a scalable, reliable, optimized infrastructure for continuous improvement of AI models • Ensure security and compliance of deployed models • Optimize model performance at runtime • Lead a team of 5-7 Software Engineers • Deliver novel, strategic solutions to improve fairness, security, and innovation in AI models • Influence the AI community by developing state-of-the-art solutions for responsible AI and privacy-preserving continuous learning

Requirements: • Bachelor's Degree in Computer Science, Engineering, Physics, Mathematics or equivalent • 10+ years of software engineering experience • 3+ years managing/mentoring engineers • Proficiency in Python and ML frameworks (TensorFlow, Keras, PyTorch) • Advanced knowledge of cloud environments (AWS, Azure, Oracle, or Google Cloud) • Excellent problem-solving and software design skills • Understanding of system architecture • Strong communication skills

Benefits: • Competitive salary and 401k with employer match • Discretionary paid time off • Paid parental leave • Medical, Dental, Vision plans • Fitness Programs • Emotional & Mental Wellness support • Learning & Development programs

Axon values diversity and is committed to building inclusive teams. They encourage applications from candidates who may not meet every qualification but are excited about the role and their mission to Protect Life.