Principal Machine Learning Engineer
Axon · Seattle, WA · 2 wk ago
HybridEngineering$177k–$283k/yrFull-time
Your Impact
Are you passionate about AI? Do you love engineering solutions enabling fast, cutting-edge science? Are you eager to contribute to building products impacting the world for the greater good?
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 You Bring
- 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.
Preferred
- 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.
Benefits that Benefit You
- 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)
- And yes, we have snacks in our offices