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. Like our products, we work better together. We connect with candor and care, seeking out diverse perspectives from our customers, communities and each other. Life at Axon is fast-paced, challenging and meaningful. Here, you'll take ownership and drive real change. Constantly grow as you work hard for a mission that matters at a company where you matter.

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. Collaborating closely with scientists and engineers, you will 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 at Axon across the board. As part of a multidisciplinary team, you will 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. They will bring in outstanding hands-on management skills. They will understand the AI lifecycle and be capable of supporting scientists along the development of state-of-the-art AI solutions, thus contributing to transforming the public safety space.

Key Responsibilities:

• Architect, develop and manage a scalable, reliable, optimized infrastructure for AI models
• Ensure security and compliance of deployed models
• Optimize model performance at runtime
• Lead a team of 5-7 Software Engineers
• Deliver strategic solutions for model fairness, security, and innovation
• Influence the AI community with responsible AI and privacy-preserving solutions

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, Google Cloud)
• Excellent problem-solving and software design skills
• Understanding of system architecture

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

Join Axon and be part of a team that's revolutionizing public safety technology while working on cutting-edge AI solutions!