Axon · 5 hours ago
Senior Machine Learning Engineer II
Washington, United States
Full-time
Hybrid
Senior Level, Lead/Staff
$157K/yr - $251K/yr
10+ years expAxon is on a mission to Protect Life, focusing on critical safety and justice issues through their devices and cloud software. The Senior Machine Learning Engineer II will architect and implement AI solutions for Axon products, collaborating with scientists and device engineers to enhance AI capabilities.
ElectronicsGovTechHardwareSoftware
Responsibilities
Architect and develop secure, privacy-preserving, on device solutions to enable the continuous improvement of existing AI models
Collaborate with scientists in architecting and implementing state-of-the-art edge distributed training techniques
Implement on device monitoring solutions used for continuous model improvement
Implement innovative model compression solutions to enable AI at the edge
Impact the team by bringing your own expertise and deep knowledge of the state-of-the-art to introduce new techniques leading to tangible impact in terms of model fairness, performance, and platform scalability
Qualification
Required
Bachelor's Degree in Computer Science, Engineering, Physics, Mathematics or an equivalent highly technical field
10+ years of software engineering experience and a proven track record of successfully architecting and maintaining large-scale distributed platforms
Experience with AI on chips, on device model deployment and management
Proficiency in python, C++, familiarity with ML frameworks such as TensorFlow, or PyTorch
Advanced knowledge and hands-on experience with on chips development
Excellent problem solving skills and ability to dive into system architecture, design, performance metrics, code, test plans, project plans, deployments and operations
Comfort communicating and interacting with scientists, engineers and product managers
Preferred
Master's Degree in Computer Science, Engineering, Physics, Mathematics or an equivalent highly technical field
Hands-on experience in solving Computer Vision or Natural Language Understanding problems in a business setting
Familiarity with responsible AI, de-biasing, model encryption and de-identification techniques
Benefits
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
Employee Resource Groups (ERGs)
And yes, we have snacks in our offices
Company
Axon
Axon provides electronic control devices to law enforcement and corrections agencies.
1001-5000 employees
H1B Sponsorship
Axon has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (3)
2024 (3)
2023 (7)
2020 (2)
Funding
Current Stage
Public CompanyTotal Funding
$2B2025-03-05Post Ipo Debt· $1.75B
2018-08-07Post Ipo Equity· $246M
2001-06-07IPO
Leadership Team
Jay Reitz
SVP Software Engineering
Anthony Baldoni
VP Strategic Initiatives
Recent News
2026-01-09
The Motley Fool
2026-01-08
Columbia West
2026-01-06
Company data provided by crunchbase