Sr. 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.
• 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 are a fast-paced, mission-driven company where engineers take ownership and drive real-world change.