Axon · 2 hours ago
Senior Machine Learning Engineer
Sterling, Virginia, United States
Full-time
Hybrid
Senior Level
5+ years expAxon is on a mission to Protect Life, focusing on critical safety and justice issues with their devices and cloud software. As a Senior Machine Learning Engineer, you will play a pivotal role in advancing airspace security by developing algorithms to detect and classify flying objects using cutting-edge technology.
ElectronicsGovTechHardwareSoftware
Responsibilities
Design and implement high-performance C++ software that runs computer vision and tracking algorithms in real time on edge devices
Work closely with computer vision / self-supervised learning engineers to integrate their models into production pipelines, including pre/post-processing, I/O, and system orchestration
Build and optimize multithreaded and parallel processing pipelines for ingesting, synchronizing, and processing data from a networked system of cameras
Implement and tune CUDA kernels and GPU-accelerated components to maximize throughput and minimize latency for inference, tracking, and search
Design robust data structures and memory management strategies for handling large volumes of video, sensor, and metadata streams under tight compute and power constraints
Profile and optimize code using tools such as perf, valgrind, nvprof / Nsight, and similar to identify bottlenecks and improve CPU/GPU utilization
Collaborate with simulation and CV teams to deploy and evaluate algorithms in realistic test scenarios, including fault handling and performance monitoring
Develop clean, well-tested, and well-documented C++ libraries and services that can be reused across products and future airspace applications
Contribute to system-level architecture decisions, including inter-process communication, scheduling, resource allocation, and deployment strategies on edge platforms
Qualification
Required
5+ years of professional experience in modern C++ (C++14/17 or later), with strong object-oriented and generic programming skills
Deep understanding of multithreading and concurrency (threads, thread pools, locks, lock-free structures, atomics, futures, async patterns) and experience building robust, concurrent systems
Hands-on experience with parallel processing frameworks or patterns (SIMD, task-based parallelism, GPU offload, or similar) for real-time or high-throughput applications
Strong command of data structures and algorithms, and the ability to choose and implement the right structures for performance-critical, memory-constrained environments
Proven experience with memory management and performance optimization in C++ (stack vs heap, custom allocators, cache-aware design, avoiding fragmentation, RAII, move semantics)
Practical experience with CUDA (or similar GPU programming frameworks): writing kernels, managing GPU memory, optimizing for occupancy and bandwidth, and integrating with C++ codebases
Strong debugging and profiling skills across CPU and GPU, and a methodical approach to benchmarking and regression testing
Excellent collaboration and communication skills, with a track record of working closely with research or ML teams to move algorithms from prototype to production
Preferred
Experience integrating machine learning or computer vision inference engines (e.g., TensorRT, OpenVINO, ONNX Runtime) is a strong plus
Familiarity with Linux-based development (build systems like CMake, unit testing frameworks, containerization and/or cross-compilation for edge devices)
Benefits
Competitive salary and 401k with employer match
Discretionary paid time off
Paid parental leave for all
Medical, Dental, Vision plans
Fitness Programs
Emotional & Mental Wellness support
Learning & Development programs
And yes, we have snacks in our offices
Company
Axon
Axon provides electronic control devices to law enforcement and corrections agencies.
1001-5000 employees
H1B Sponsorship
Axon has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (3)
2024 (3)
2023 (7)
2020 (2)
Funding
Current Stage
Public CompanyTotal Funding
$2B2025-03-05Post Ipo Debt· $1.75B
2018-08-07Post Ipo Equity· $246M
2001-06-07IPO
Leadership Team
Jay Reitz
SVP Software Engineering
Anthony Baldoni
VP Strategic Initiatives
Recent News
2026-01-09
The Motley Fool
2026-01-08
Columbia West
2026-01-06
Company data provided by crunchbase