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.
• Deep expertise in building distributed systems or scaling 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 connect with candor and care, seeking out diverse perspectives to build a safer, better world.