Machine Learning Engineer II

Axon develops devices and cloud software for public safety, focusing on protecting life through innovative technology solutions.
$140,000 - $260,000
Machine Learning
Mid-Level Software Engineer
Remote
1,000 - 5,000 Employees
5+ years of experience
AI · Enterprise SaaS

Description For Machine Learning Engineer II

Axon is seeking a Machine Learning Engineer II to join their mission of protecting life through innovative technology. This role combines software engineering expertise with machine learning capabilities, focusing on building and maintaining AI infrastructure that powers various Axon products. The position offers an opportunity to work on cutting-edge AI applications ranging from computer vision to speech recognition and NLU.

The ideal candidate will be responsible for architecting and implementing platforms used by Axon scientists to transform public safety. Working in a multidisciplinary team, you'll support the training, evaluation, testing, and deployment of machine learning models across Axon's product line including Fleet, Axon Body, and Axon Air. The role requires expertise in managing large-scale distributed platforms for secure and compliant research in the cloud.

Axon offers a competitive compensation package with base pay ranging from $140,000 to $260,000, plus bonus and stock awards. The company provides comprehensive benefits including medical insurance, flexible working hours, and various perks such as free lunch and gym access. This position offers unique opportunities to impact public safety through technology while working with state-of-the-art AI solutions.

The role requires 5+ years of software engineering experience, strong Python skills, and expertise with ML frameworks. You'll be working in a fast-paced, meaningful environment where you can drive real change while contributing to society's critical safety and justice issues. The position offers both technical challenges and the satisfaction of working on products that make a difference in public safety.

Last updated 5 days ago

Responsibilities For Machine Learning Engineer II

  • Architect, develop and manage scalable infrastructure for training and testing AI models
  • Deliver strategic solutions to improve model fairness and security
  • Develop state-of-the-art solutions for responsible AI
  • Support training and deployment of machine learning models
  • Collaborate with scientists and engineers on AI capabilities

Requirements For Machine Learning Engineer II

Python
Kubernetes
  • Bachelor's Degree in Computer Science, Engineering, Physics, Mathematics or equivalent
  • 5+ years of software engineering experience
  • Proficiency in Python and ML frameworks (TensorFlow, Keras, PyTorch)
  • Advanced knowledge of cloud environments (AWS, Azure, or Google Cloud)
  • Excellent problem solving and software design skills
  • Understanding of system architecture
  • Strong communication skills

Benefits For Machine Learning Engineer II

Medical Insurance
  • Competitive salary and benefits
  • Medical insurance for employee and family
  • Free lunch
  • Gym access
  • Phone stipend
  • Parking
  • Flexible working hours
  • Training opportunities
  • US rotation opportunities

Interested in this job?

Jobs Related To Axon Machine Learning Engineer II

Machine Learning Engineer II

Machine Learning Engineer role at Axon focusing on developing AI solutions for public safety products including speech recognition, computer vision, and NLU capabilities.

Software Development Engineer | Amazon Delivery Experience (DEX) | Machine Learning, DEX AI Regionalization Engineering

Software Development Engineer position at Amazon's Delivery Experience team, focusing on ML-based solutions for delivery optimization.

Software Development Engineer II, ML_AI

AWS SDE II role focusing on building next-gen AI platform for large-scale deep learning, working with LLMs and distributed systems at Amazon's cloud division.

Software Dev. Engineer, Alexa Analytics

Software Development Engineer position at Amazon's Alexa Analytics team, combining software engineering with ML expertise to build scalable analytics applications and improve Alexa's predictive capabilities.

Machine Learning Engineer II, Special Projects

Machine Learning Engineer II position at Amazon's Special Projects team, focusing on Generative AI and LLMs, offering competitive compensation and comprehensive benefits.