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 proficiency in Python and C++.
• Extensive experience with modern ML frameworks such as PyTorch or TensorFlow.
• Proven track record of delivering end-to-end systems from concept to production.

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.
• Strong system design skills across cloud and edge environments.

Benefits

• Competitive salary and 401k with employer match.
• Discretionary paid time off and paid parental leave.
• Comprehensive 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 an ecosystem of devices and cloud software. We work at the intersection of AI, systems, and product to build scalable, high-impact solutions that make the world a safer place.