Taro Logo

Machine Learning Engineer II

Axon develops technology and products for law enforcement, military, and civilians to promote safety and justice.
Scottsdale, AZ, USA
$161,000 - $309,000
Machine Learning
Senior Software Engineer
Remote
1,000 - 5,000 Employees
8+ years of experience
AI
This job posting may no longer be active. You may be interested in these related jobs instead:

Description For Machine Learning Engineer II

Axon is seeking a Machine Learning Engineer II to join their mission of protecting life through innovative technology solutions. This role combines cutting-edge AI development with meaningful impact in public safety. As a ML Engineer, you'll be instrumental in architecting and implementing platforms for AI capabilities across Axon's product line, including Fleet, Axon Body, and Axon Air. The position involves working with diverse AI applications from computer vision to speech recognition and NLU/LLMs. You'll collaborate with scientists and engineers to develop state-of-the-art AI solutions while ensuring secure and compliant research in the cloud. The role offers competitive compensation ($161,000-$309,000), comprehensive benefits, and the opportunity to work on technology that directly impacts public safety. Axon provides a dynamic, mission-driven environment where you can apply advanced ML engineering skills to create solutions that make a difference in law enforcement and community safety. The position requires 8+ years of experience and deep expertise in ML frameworks, cloud environments, and distributed systems, making it ideal for experienced ML engineers passionate about combining technical innovation with social impact.

Last updated 3 months ago

Responsibilities For Machine Learning Engineer II

  • Architect solutions to train and evaluate models on a distributed architecture
  • Optimize the evaluation and training of models within Axon
  • Architect novel, strategic distributed solutions for continuous model improvement
  • Deliver novel, strategic solutions to accelerate innovation
  • Influence to the AI community by developing state-of-the-art solutions for privacy-preserving distributed model training

Requirements For Machine Learning Engineer II

Python
Kubernetes
  • Bachelor's Degree in Computer Science, Engineering, Physics, Mathematics or an equivalent highly technical field
  • 8+ years of software engineering experience
  • Proficiency in python, familiarity with ML frameworks such as TensorFlow, Keras, PyTorch
  • Advanced knowledge and hands-on experience with at least one cloud environment
  • Experience with CI/CD solutions in the context of MLOps including automation with IaC
  • Excellent problem solving and software design skills
  • Comfort communicating and interacting with scientists, engineers and ML/product managers

Benefits For Machine Learning Engineer II

Medical Insurance
Dental Insurance
Vision Insurance
401k
  • Competitive salary and benefits including a great medical insurance plan
  • Free lunch, gym, phone stipend, parking
  • Flexible working hours
  • Opportunities for training and rotations in the US
  • An awesome office/working environment with the latest technologies

Interested in this job?