Senior Machine Learning Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

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
check
Growth Opportunities
check
H1B Sponsor Likelynote

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

Machine LearningPythonCloud ArchitectureC++LinuxInfrastructure-as-codeLLMOpsSoftware DesignProblem SolvingCommunication Skills

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 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 Company
Total Funding
$2B
2025-03-05Post Ipo Debt· $1.75B
2018-08-07Post Ipo Equity· $246M
2001-06-07IPO

Leadership Team

leader-logo
Jay Reitz
SVP Software Engineering
linkedin
leader-logo
Anthony Baldoni
VP Strategic Initiatives
linkedin
Company data provided by crunchbase