Senior Machine Learning Engineer

Join Axon and be a Force for Good.

At Axon were on a mission to Protect Life. Were explorers pursuing societys most critical safety and justice issues with our ecosystem of devices and cloud software. Like our products we work better together. We connect with candor and care seeking out diverse perspectives from our customers communities and each other.

Life at Axon is fast-paced challenging and meaningful. Here youll take ownership and drive real change. Constantly grow as you work hard for a mission that matters at a company where you matter.

Your Impact

Are you eager to contribute to building AI products impacting the world for the greater good

As an ML Engineer at Axon you will contribute to developing AI solutions transforming the public safety space and supporting our companys mission: accelerate justice protect truth and save lives. Working closely with other ML engineers and scientists you will enable new AI capabilities for Axons products (Fleet Draft One Axon Air VR and more) by supporting the training evaluation testing and deployment of machine learning models on Axon devices and cloud.

As part of a multidisciplinary team you will be exposed to a wide range of AI domains and applications: from computer vision to speech recognition and natural language understanding.

What Youll Do

Location: Remotely from US or Hybrid from our office in Seattle Washington.
Reports to: Manager Machine Learning Engineering

• Collaborate with scientists and product managers to build proof-of-concepts (POCs) contributing to shaping the Axon of tomorrow.
• Architect and develop secure privacy-preserving solutions to enable the continuous improvement of existing AI models.
• Architect platforms that accelerate research and AI product development.
• Collaborate with scientists in architecting and implementing state-of-the-art training techniques.
• Set high standards for ethical and responsible AI development.

What You Bring

• Bachelors Degree in Computer Science Engineering Electronics Mathematics or an equivalent highly technical field.
• 6 years of software engineering experience and a proven track record of successfully deploying AI models to the cloud.
• Experience with Infrastructure-as-code and cloud architecture.
• Proficiency in Python and C familiarity with ML frameworks such as TensorFlow or PyTorch.
• Advanced knowledge and hands-on experience with Linux.
• Excellent problem solving skills and ability to dive deep into system architecture.
• Excellent software design skills.
• Comfort communicating and interacting with scientists engineers and product managers.

What You Bring

• Masters Degree/PhD in Computer Science Engineering Electronics Mathematics or an equivalent highly technical field.
• Experience with LLMOps - evaluation monitoring quantization teacher-learner etc.).
• Hands-on experience with fine-tuning large parameter models.
• Familiarity with model encryption and privacy preserving AI techniques.

Benefits that Benefit You

• Competitive salary and 401k with employer match
• Discretionary paid time off
• Paid parental leave for all
• Medical Dental Vision plans
• Fitness Programs
• Emotional & Mental Wellness support
• Learning & Development programs
• And yes we have snacks in our offices

Benefits listed herein may vary depending on the nature of your employment and the location where you work

The Pay: Axon is a total compensation company meaning compensation is made up of base pay bonus and stock awards. The starting base pay for this role is between USD 150750 in the lowest geographic market and USD 221000 in the highest geographic market. The actual base pay is dependent upon many factors such as: level function training transferable skills work experience business needs geographic market and often a combination of all these factors. Our benefits offer an array of options to help support you physically financially and emotionally through the big milestones and in your everyday life. To see more details on our benefits offerings please visit meet every single requirement Thats ok. At Axon we Aim Far. We think big with a long-term view because we want to reinvent the world to be a safer better place. We are also committed to building diverse teams that reflect the communities we serve.

Studies have shown that women and people of color are less likely to apply to jobs unless they check every box in the job description. If youre excited about this role and our mission to Protect Life but your experience doesnt align perfectly with every qualification listed here we encourage you to apply anyways. You may be just the right candidate for this or other roles.

Important Notes

The above job description is not intended as nor should it be construed as exhaustive of all duties responsibilities skills efforts or working conditions associated with this job. The job description may change or be supplemented at any time in accordance with business needs and conditions.

Some roles may also require legal eligibility to work in a firearms environment.

We collect personal information from applicants to evaluate candidates for employment. You may request access deletion or exercise other CCPA rights at or via our Axon Privacy Web Form. For more information please see the Your California Privacy Rights section of our Applicant and Candidate Privacy Notice.

Axons mission is to Protect Life and is committed to the well-being and safety of its employees as well as Axons impact on the environment. All Axon employees must be aware of and committed to the appropriate environmental health and safety regulations policies and procedures. Axon employees are empowered to report safety concerns as they arise and activities potentially impacting the environment.

We are an equal opportunity employer that promotes justice advances equity values diversity and fosters inclusion. Were committed to hiring the best talent regardless of race creed color ancestry religion sex (including pregnancy) national origin sexual orientation age citizenship status marital status disability gender identity genetic information veteran status or any other characteristic protected by applicable laws regulations and ordinances and empowering all of our employees so they can do their best work. If you have a disability or special need that requires assistance or accommodation during the application or the recruiting process please email . Please note that this email address is for accommodation purposes only. Axon will not respond to inquiries for other purposes.

Required Experience:

Senior IC