ML Engineer

Axon develops devices and cloud software for safety and justice, creating an ecosystem of products to Protect Life.
$161,000 - $309,000
Machine Learning
Staff Software Engineer
Remote
8+ years of experience
AI · Enterprise SaaS

Description For ML Engineer

Axon is seeking a Machine Learning Engineer to join their mission of protecting life through innovative safety and justice solutions. This role combines AI expertise with real-world impact, focusing on developing and implementing platforms for AI capabilities across Axon's product line including Fleet, Axon Body, and Axon Air. The position offers exposure to cutting-edge domains from computer vision to speech recognition, NLU, and LLMs.

The ideal candidate will bring 8+ years of software engineering experience, with expertise in distributed platforms and AI lifecycle management. You'll work in a multidisciplinary team, architecting solutions for model training and evaluation, while contributing to state-of-the-art privacy-preserving distributed model training.

Axon offers a competitive compensation package ranging from $161,000 to $309,000 base salary, plus bonus and stock awards. The company provides comprehensive benefits including medical insurance, flexible working hours, and unique opportunities like police ride-alongs to understand technology's impact in real-world scenarios.

The role combines technical excellence with meaningful impact, allowing you to contribute to society's critical safety and justice issues. Axon's culture emphasizes candor, care, and diverse perspectives, creating an environment where innovation thrives and personal growth is continuous. This position offers the rare opportunity to work on cutting-edge AI technology while directly contributing to public safety and social justice initiatives.

Last updated 2 days ago

Responsibilities For ML Engineer

  • 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 the AI community by developing state-of-the-art solutions for privacy-preserving distributed model training

Requirements For ML Engineer

Python
  • Bachelor's Degree in Computer Science, Engineering, Physics, Mathematics or equivalent technical field
  • 8+ years of software engineering experience
  • Proficiency in Python, familiarity with ML frameworks (TensorFlow, Keras, PyTorch)
  • Advanced knowledge of cloud environments (AWS, Azure, Oracle, or Google Cloud)
  • Experience with CI/CD solutions in MLOps including IaC
  • Excellent problem solving and software design skills
  • Strong communication skills

Benefits For ML Engineer

Medical Insurance
  • Competitive salary and benefits
  • Great medical insurance plan for you and your family
  • Free lunch
  • Gym access
  • Phone stipend
  • Free parking
  • Flexible working hours
  • Training opportunities
  • US rotation opportunities
  • Latest technologies
  • Police ride-along opportunities

Interested in this job?

Jobs Related To Axon ML Engineer

ML Engineer Manager

Lead ML engineering team at Axon, developing AI solutions for public safety with focus on responsible AI and privacy-preserving technologies.

ML Engineer Manager

Lead ML engineering team at Axon, developing AI solutions for public safety with focus on responsible AI and privacy-preserving technologies.

Manager, Machine Learning Engineering

Lead AI innovation at Axon as a Machine Learning Engineering Manager, developing cutting-edge solutions for public safety.

Manager, Machine Learning Engineering

Lead ML Engineering team at Axon, architecting AI platforms for public safety products. 8+ years experience required.

Senior Manager, Machine Learning Engineering

Senior Manager, Machine Learning Engineering at Axon: Lead AI innovation in public safety, managing a team to develop cutting-edge ML solutions.