Axon is hiring a Principal Machine Learning Engineer to architect and implement the platform that enables Dedrone scientists to deliver safe, secure, and reliable AI features for counter-UAS devices. You will enable new AI capabilities by collaborating closely with scientists and software engineers across domains from Multi-modal Sensor Fusion to Real Time Systems with distributed edge compute.
What You'll Do
- Architect and develop secure, privacy-preserving, on device solutions to enable the continuous improvement of existing AI models.
- Collaborate with scientists in architecting and implementing state-of-the-art edge distributed training techniques.
- Implement on device monitoring solutions used for continuous model improvement.
- Implement innovative model compression solutions to enable AI at the edge.
- Impact the team by bringing your own expertise and deep knowledge of the state-of-the-art to introduce new techniques leading to tangible impact in terms of model fairness, performance, and platform scalability.
What We're Looking For
- Bachelor’s Degree in Computer Science, Engineering, Physics, Mathematics or an equivalent highly technical field.
- 13+ years of software engineering experience and a proven track record of successfully architecting and maintaining large-scale distributed platforms.
- Experience with AI on chips, on device model deployment and management.
- Proficiency in python, C++, familiarity with ML frameworks such as TensorFlow, or PyTorch.
- Advanced knowledge and hands-on experience with on chips development.
- Excellent problem solving skills and ability to dive into system architecture, design, performance metrics, code, test plans, project plans, deployments and operations.
- Comfort communicating and interacting with scientists, engineers and product managers.
Nice to Have
- Master’s Degree or PhD in Computer Science, Engineering, Physics, Mathematics or an equivalent highly technical field.
- Hands-on experience in solving Computer Vision or Natural Language Understanding problems in a business setting.
- Familiarity with responsible AI, de-biasing, model encryption and de-identification techniques.
Technical Stack
- Python
- C++
- TensorFlow
- PyTorch
Team & Environment
You'll join a multidisciplinary team collaborating with scientists and software engineers, reporting to the VP, Dedrone Engineering.
Benefits & Compensation
- 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
- Employee Resource Groups (ERGs)
- Snacks in our offices
- Compensation Range: $177,000—$283,200 USD
Work Mode
This is a hybrid role based in Seattle, WA.
We are an equal opportunity employer that promotes justice, advances equity, values diversity and fosters inclusion. We’re committed to hiring the best talent — regardless of race, creed, color, ancestry, religion, sex (including pregnancy), national origin, sexual orientation, age, citizenship status, marital status, disability, gender identity, genetic information, veteran status, or any other characteristic protected by applicable laws, regulations and ordinances.
