Latest AI JobsBrowse Jobs
Axon logo

Principal Machine Learning Engineer

AxonSeattle, Washington, United States2024 Dedrone R&D
Apply on company site
Salary
Not listed
Job type
Full-time
Posted
10 Mar 2026
Status
Open · checked 7d ago

About this role

The Principal Machine Learning Engineer position at Axon in Seattle, Washington is a hybrid role focused on building edge AI platforms for counter-UAS devices. This opportunity suits an experienced software engineering professional with a strong background in distributed systems and artificial intelligence.

Role Overview

As a Principal ML Engineer within the Dedrone R&D department at Axon, the professional will architect and implement platforms that allow scientists to bring secure and reliable AI features to counter-UAS technology. Reporting to the Director of Artificial Intelligence, this role operates on a hybrid schedule out of the Seattle, Washington office, requiring onsite collaboration from Tuesday through Friday with remote flexibility on Mondays.

Key Responsibilities

Architect and build secure, privacy-preserving, on-device solutions for continuous machine learning model improvement.
Partner with scientists to design and implement state-of-the-art edge distributed training techniques.
Build on-device monitoring solutions to support ongoing model optimization.
Introduce innovative model compression methods to facilitate advanced AI functionality at the edge.
Enhance platform scalability, model fairness, and overall performance using deep domain expertise.

Requirements

Bachelor’s Degree in Computer Science, Engineering, Physics, Mathematics, or a related technical discipline.
Minimum of 13 years of software engineering experience with a strong track record of building and maintaining large-scale distributed platforms.
Hands-on experience with AI on chips, on-device model deployment, and lifecycle management.
Proficiency in programming languages such as Python and C++, alongside familiarity with machine learning frameworks like TensorFlow or PyTorch.
Excellent problem-solving capabilities spanning system architecture, design, metrics, code, and testing operations.
Strong communication skills to interact effectively with scientists, product managers, and software engineers.

Nice to Have

Master’s Degree or PhD in Computer Science, Engineering, Physics, Mathematics, or a similarly technical field.
Practical experience solving Natural Language Understanding or Computer Vision challenges in a commercial environment.
Familiarity with responsible AI principles, model encryption, de-biasing, and de-identification methods.

Skills and Tools

Python
C++
TensorFlow
PyTorch
Edge AI and On-Device Deployment
Computer Vision and Sensor Fusion

Salary and Benefits

Base pay range of $177,000 to $283,200 USD, supplemented by bonus and stock awards as part of Axon's total compensation approach.
401(k) plan with employer match.
Comprehensive medical, dental, and vision insurance packages.
Discretionary paid time off and paid parental leave for all employees.
Fitness programs, emotional and mental wellness support, and learning and development opportunities.
Access to employee resource groups and office snacks.

About the Company

Axon is on a mission to Protect Life, developing an ecosystem of hardware devices and cloud software aimed at addressing critical safety and justice issues. Operating in fast-paced and challenging environments, the company values diverse perspectives, collaboration, and meaningful work that drives real change in public safety, enterprise, and defense domains.

How to Apply

Interested candidates should visit the official Axon careers page to submit an application for this position.

Skills and tech stack

Questions candidates ask

What is the location and work schedule for this role?

The role is based in Seattle, Washington, and follows a hybrid schedule. Team members work onsite from Tuesday to Friday, with the option to work remotely on Mondays.

What is the salary range for the Principal Machine Learning Engineer?

The base pay range is between $177,000 and $283,200 USD, which may vary based on factors such as experience, level, and geographic market, alongside bonus and stock awards.

What qualifications are required for this position?

Candidates need a Bachelor's Degree in a technical field like Computer Science or Engineering, at least 13 years of software engineering experience, proficiency in Python and C++, familiarity with frameworks like PyTorch or TensorFlow, and experience with on-device model deployment.

Are there preferred qualifications for applicants?

Yes, preferred qualifications include a Master’s Degree or PhD in a technical field, hands-on computer vision or natural language understanding experience, and familiarity with responsible AI and model encryption.

What benefits does Axon offer?

Axon offers medical, dental, and vision plans, a 401(k) with employer match, discretionary paid time off, paid parental leave, fitness programs, and mental wellness support.

Ready to apply?

The application is on the official Axon careers page.

Apply on company site