Senior Machine Learning Engineer

As a Senior Machine Learning Engineer at Axon, you will play a pivotal role in advancing airspace security through the development of cutting-edge camera systems and machine learning algorithms. Your work will focus on detecting, tracking, and classifying diverse flying objects in complex airspace environments, contributing directly to Axon's mission to protect life and ensure public safety.

In this role, you will design and implement high-performance C++ software that runs computer vision and tracking algorithms in real time on edge devices. Collaborating closely with computer vision and self-supervised learning engineers, you will integrate models into production pipelines, optimize multithreaded and parallel processing pipelines, and implement GPU-accelerated components to maximize throughput and minimize latency.

The ideal candidate will have over five years of professional experience in modern C++ (C++14/17 or later), with strong object-oriented and generic programming skills. A deep understanding of multithreading and concurrency, hands-on experience with parallel processing frameworks, and practical experience with CUDA or similar GPU programming frameworks are essential. Familiarity with integrating machine learning or computer vision inference engines and Linux-based development is also highly desirable.

Axon offers a competitive salary and a comprehensive benefits package, including discretionary paid time off, paid parental leave, and medical, dental, and vision plans. The company fosters a collaborative and innovative work environment, emphasizing in-person collaboration to fuel innovation and meaningful teamwork. This role is based out of Axon's Sterling, Virginia office and follows a hybrid schedule, with onsite work from Tuesday through Friday and remote work on Mondays.