Axon · 2 hours ago
Senior Machine Learning Scientist
Axon is on a mission to Protect Life, focusing on critical safety and justice issues through their ecosystem of devices and cloud software. They are seeking a Senior Machine Learning Scientist to advance AI applications in Cloud, Devices, and Robotics, collaborating on innovative solutions in Large Language Models and Computer Vision.
ElectronicsGovTechHardwareSoftware
Responsibilities
Own one or more key technical areas across LLM, MLLM, CV product portfolio
Provide technical leadership to junior scientists, guiding the transition of R&D concepts into impactful Axon product feature
Research and develop cutting-edge techniques in LLM, MLLMs, GenAI, and Computer Vision across cloud, devices and sensors based data sources
Design and implement efficient and scalable MLLM models for inference and analysis of multimodal data
Explore novel approaches to address challenges in NLP, NLU, Object Detection, Object Recognition, Object Tracking, Segmentation, and Scene Understanding
Optimize AI models, algorithms for performance, memory footprint, and energy efficiency to meet the requirements of resource-constrained devices
Join force with MLEs or firmware or hardware engineers to leverage hardware accelerators and optimize algorithms for specific hardware architectures
Evaluate the performance of LLM, MLLM, CV models using real-world datasets and design experiments to validate their effectiveness
Stay up-to-date with the latest research trends and advancements in computer vision, machine learning, and deep learning, MLLMs, GenAI and integrate relevant findings into our projects
Contribute to patent disclosures, academic publications, and technical documentation to share insights and findings with the broader community
Experience coach and mentor junior scientists
Qualification
Required
PhD and with +5 years for ML Scientist, +8 years for Sr. ML Scientist, +10 years for Principal ML Scientist experience in Computer Science or a related field with a focus on LLM, MLLMs, Computer Vision, GenAI
Proven track record of research excellence in LLM, MLLM, Computer Vision, Robotics Perception, GenAI, demonstrated through publications in top-tier conferences or journals
Strong proficiency in programming languages such as Python, C/C++, experience with deep learning frameworks such as TensorFlow, PyTorch, or Keras and experience with ROS or robotic operational system
Drive one or more phases of the ML development lifecycle: shape datasets, investigate modeling approaches and architectures, train/evaluate/tune models and implement the end-to-end training pipeline
Leverage state-of-the-art research to deliver high quality models enabling multiple AI projects at scale
Contribute back to the research community via academic publications, tech blogs, open-source code and contributing to internal/external AI challenges
Excellent problem-solving skills, analytical thinking, and the ability to work independently as well as collaboratively in a team environment
Strong communication skills and the ability to effectively present complex technical concepts to both technical and non-technical audiences
Preferred
Experience in developing computer vision algorithms for resource-constrained devices such as mobile phones, IoT devices, or embedded systems is highly desirable
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
Government Technology US
2026-02-07
2026-01-09
The Motley Fool
2026-01-08
Company data provided by crunchbase