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 18 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

Sr. Software Development Engineer, Demand Science Optimization (DSO)

Senior Software Engineering role at Amazon focusing on machine learning and big data analytics for device demand forecasting and supply chain optimization.

Delivery Consultant - Machine Learning Engineer, WWPS ProServe

Senior ML Engineering role at AWS Professional Services, focusing on implementing machine learning solutions for enterprise customers using AWS cloud services.

Senior Software Engineer, LLM Inference

Senior Software Engineer position at NVIDIA focusing on LLM Inference development, requiring expertise in C++, deep learning, and AI technologies.

Generative AI Engineer - Model Optimization & Evaluation

Senior AI Engineering role focused on optimizing and evaluating transformer-based models, requiring expertise in model compression, quantization, and deployment across various computing environments.

Sr. Machine Learning Engineer

Senior Machine Learning Engineer role at Adobe focusing on developing and deploying ML models and pipelines for content generation, offering competitive compensation $142K-$257K.