Jobs · Engineering · Washington

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

Similar jobs