Staff AI Embedded Software Engineer

Responsibilities

• Define and advance embedded software architectures for connected device products, including AI-enabled systems and cloud-assisted workflows.
• Lead the technical direction for AI capabilities across multiple product lines, including model training, data strategy, and deployment.
• Partner with research and product teams to integrate emerging AI approaches like foundation models and multimodal systems.
• Establish Axon-wide standards for embedded software and AI system design, focusing on reliability, safety, and scalability.
• Mitigate risks associated with AI systems, such as model failure modes and data drift, to ensure reliable real-world behavior.
• Provide executive-level guidance and mentorship to enhance the technical decision-making of embedded software engineering teams.

Requirements

• 12+ years of professional software development experience with expertise in C/C++, Go, Python, or comparable systems languages.
• Deep expertise in embedded systems architecture, firmware integration, and device-level software engineering.
• Hands-on experience deploying and optimizing AI inference workloads on constrained edge platforms (MCUs, SoCs, NPUs).
• Proven experience designing, training, and operating machine learning models at scale, including ownership of data pipelines.
• Practical experience with large-scale AI systems, including fine-tuning or integrating foundation models and LLMs.
• 7+ years of experience mentoring senior engineers and leading complex, strategic engineering initiatives.

Preferred Qualifications

• Experience with networking and distributed system concepts relevant to connected and AI-enabled devices.
• Proven track record of resolving system-wide challenges in performance, security, and safety within mission-critical environments.

Benefits

• Competitive salary and 401k with employer match
• Discretionary paid time off and paid parental leave
• Medical, Dental, and Vision plans
• Fitness programs and emotional/mental wellness support
• Learning and development programs

About the Company

Axon is on a mission to Protect Life. We build an ecosystem of devices and cloud software to address society’s most critical safety and justice issues. We work in a fast-paced, meaningful environment where employees take ownership and drive real change.