Axon · 2 days ago
Senior Machine Learning Engineer
Axon is on a mission to Protect Life through innovative devices and cloud software. As a Senior Machine Learning Engineer, you will develop AI solutions that enhance public safety, collaborating with a multidisciplinary team to enable new AI capabilities for Axon’s products.
ElectronicsGovTechHardwareSoftware
Responsibilities
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
Qualification
Required
Bachelor's 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
Preferred
Master's 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
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
Company
Axon
Axon provides electronic control devices to law enforcement and corrections agencies.
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
Recent News
2025-12-29
2025-12-08
Company data provided by crunchbase