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

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 5 days 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?

Jobs Related To Axon Machine Learning Engineer II

Senior MLOps Engineer (Na)

Senior MLOps Engineer position at Intellectsoft focusing on AI-powered platform development, requiring 5+ years of experience in software engineering and ML operations.

Senior Gen AI Software Engineer

Senior Gen AI Software Engineer position at Jobgether, developing cutting-edge AI solutions and intelligent agents for business automation, offering remote work in Canada.

Senior MLOps Engineer

Senior MLOps Engineer position at Intellectsoft, focusing on AI-powered platform development with ML operations, Python programming, and DevOps practices in a remote setting.

Senior Software Engineer - Content Understanding

Senior Software Engineer position at Roblox focusing on AI and content understanding, building scalable systems for processing and analyzing digital content across the platform.

Software Engineer (AI)

Senior AI Software Engineer position at Binance, focusing on developing autonomous AI agents and systems, requiring 3+ years of experience in AI/ML development with Python and Golang expertise.