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

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 offers an exciting opportunity to work on cutting-edge AI applications in the public safety sector, with a focus on developing and implementing platforms for AI model training and deployment.

The position involves working with a multidisciplinary team to support various AI capabilities across Axon's product line, including Fleet, Axon Body, and Axon Air. You'll be working on diverse applications ranging from computer vision to speech recognition and Natural Language Understanding. The role requires expertise in building and managing large-scale distributed platforms for secure and compliant research in the cloud.

As a Machine Learning Engineer at Axon, you'll be responsible for architecting and implementing the platform used by scientists to transform the public safety space. This involves close collaboration with scientists and engineers to enable new AI capabilities through the training, evaluation, testing, and deployment of machine learning models. The role particularly emphasizes responsible AI development, including work on model fairness, security, and innovation acceleration.

The position offers competitive compensation with a base salary range of $140,000 to $260,000, depending on location, plus bonus and stock awards. Axon provides comprehensive benefits including medical insurance, flexible working hours, and various perks such as free lunch and gym access. The company culture emphasizes innovation, collaboration, and making a positive impact on society through technology.

This role is ideal for candidates who are passionate about AI and want to contribute to meaningful products that help protect life. The position offers unique opportunities to work with real-world applications of AI in public safety, including ride-alongs with police officers to understand technology usage in the field. Axon's commitment to diversity and inclusion means they welcome candidates who might not meet every qualification but are passionate about the mission and have the core technical skills needed to succeed.

Last updated 5 days ago

Responsibilities For Machine Learning Engineer II

  • Architect, develop and manage scalable infrastructure for training and testing AI models
  • Deliver solutions to improve model fairness and security
  • Deliver strategic solutions to accelerate innovation
  • Influence the AI community by developing state-of-the-art solutions for responsible AI

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, Oracle, 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 including medical insurance
  • 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 II position at Axon, focusing on building AI infrastructure for public safety technology, offering competitive compensation and the opportunity to impact society positively.

Software Development Engineer, GENIE - Proactive Alexa

SDE II role at Amazon's Alexa team working on automated content generation using LLMs and Generative AI, building systems that process petabytes of data and serve millions of users.

Software Development Engineer, Amazon

Software Development Engineer II position at Amazon's AI Technology team, focusing on machine learning and AI applications for enhanced shopping experiences.

Software Development Engineer - Prime Video, Linear Personalization Experience Team (LPEX)

Build AI-powered personalization systems for Prime Video's Linear TV experience, leading the development of sophisticated recommendation systems serving millions of users.

Machine Learning Engineer, Generative AI Innovation Center - Model Customization

Machine Learning Engineer position at AWS's Generative AI Innovation Center, focusing on LLM customization, model optimization, and enterprise AI solutions.