Our mission is to protect life.

We’re out to make the world a safer place by solving big problems and taking on the public safety challenges of our time. From our company's inception building the TASER device to a full suite of hardware and software solutions, we are focused on providing police agencies with the state-of-the-art devices and services they need to successfully serve and protect us. In the next few years, we're going to eliminate the burden of paperwork in policing, so officers can increase the time they spend building relationships and serving in their communities. We’ll put video at the heart of the police record so our justice system can get to the truth faster. And we won't stop innovating until the bullet is rendered obsolete.

It’s a big mission, but it’s one we’ll pursue relentlessly every single day. 

Your Impact

As a senior member of our Machine Learning Platform at Axon, you will have the opportunity to use your talents and background in software engineering and machine learning for improving public safety and eventually saving lives. Everything we build at Axon AI, including models that eventually run in the cloud or on the edge, depends on a robust, secure, and scalable ML platform. In this role, you will have the opportunity to build that platform.

Your Day-to-Day

• You will be building automation for ML models from experimentation to production on large quantities of data across distributed environments.
• You will build on the latest and greatest of ML platforms and customize them for Axon’s unique use-cases.
• You will lead architecture design and mentor junior engineers in the team.
• You will take models that have worked well on smaller datasets and scaling and optimizing our ML pipelines and models to reach their full potential on the petabytes of footage at Axon.
• You will continually read the state of the art of research, and work with research scientists to publish novel work (optional).
• You will see that generalizable parts of your work get open-sourced.

Basic Qualifications

• Bachelors degree in computer science or related fields
• 5 years of experience in the industry
• Strong background in Python
• Strong background in High Performance Computing, Cloud Infrastructures, or GPU optimization.
• Understanding of the complete workflow from collecting and annotating the appropriate data for the task at hand to training models and evaluating their performance.
• Familiarity with ML frameworks such as Tensorflow/Keras/PyTorch/Caffe
• Experience in processing and building pipelines for large datasets in a ML context

Preferred Qualifications

• Masters or PhD in ML or related fields
• Proficiency in Scala
• Strong background in ML, stats, or a related field. MS or MEng in vision, NLP, audio, stats, or ML preferred.
• 10 years of experience in the industry
• Experience training and deploying ML models in production at scale

Compensation and Benefits

• Competitive salary and 401K with employer match
• Discretionary paid time off
• Robust parental leave policy
• An award-winning office/working environment
• Ride along with real police officers in real life situations, see them use technology, get inspired
• And more...

We are an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.