Senior Full Stack Member of Technical Staff

Responsibilities

• Own end-to-end system development across data pipelines, model development, evaluation, and deployment for AI-powered products.
• Design and build scalable, production-grade systems for real-time perception, multimodal understanding, and decision-making.
• Develop and deploy models across cloud and edge environments, including resource-constrained devices.
• Translate research advances in computer vision, NLU, GenAI, and multimodal LLMs into production systems.
• Collaborate cross-functionally with MLE, hardware, firmware, and product teams to integrate AI into real-world systems.
• Optimize solutions for hardware-specific acceleration (GPU, NPU, DSP) and edge deployment.
• Lead technical direction for complex, ambiguous problems and influence system architecture across teams.

Requirements

• Master’s or PhD in Computer Science, Electrical Engineering, or a related field.
• 8-10+ years of industry experience building production systems.
• Strong programming skills in Python and C++.
• Proficiency with modern ML frameworks such as PyTorch or TensorFlow.
• Proven experience delivering end-to-end systems from concept to production.
• Expertise in building and scaling distributed systems or cloud-based ML pipelines.

Preferred Qualifications

• Experience deploying AI on Cloud (AWS, AzureML) and edge devices (mobile, embedded, robotics).
• Familiarity with hardware acceleration frameworks like TensorRT, ONNX, or SNPE.
• Experience with real-time systems and streaming data pipelines.
• Knowledge of multimodal data processing including vision, audio, text, and sensor fusion.
• Track record of technical leadership, mentoring, and driving cross-team initiatives.

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 are explorers pursuing society’s most critical safety and justice issues with our ecosystem of devices and cloud software. We build cutting-edge technology to enable intelligent perception, reasoning, and action in mission-critical environments.