Remote Senior Machine Learning Engineer

at Axon

Posted 11 hours ago 1 applied

Description:

  • Join Axon and be a part of a mission to Protect Life by addressing critical safety and justice issues through devices and cloud software.
  • As a Senior Machine Learning Engineer for Embedded AI, you will develop AI solutions that transform the public safety space and support Axon’s mission to accelerate justice, protect truth, and save lives.
  • You will work closely with other ML engineers and scientists to enable new AI capabilities for Axon’s products, including Fleet, Axon Body, Axon Air, and VR.
  • Collaborate with scientists and product managers to build proof-of-concepts (POCs) that shape the future of Axon.
  • Architect and develop secure, privacy-preserving solutions to continuously improve existing AI models.
  • Architect the next generation of the continuous deployment framework for AI on devices.
  • Optimize models for the edge to enable new AI features on existing and new Axon devices.
  • Collaborate with scientists to implement state-of-the-art training techniques.

Requirements:

  • A Bachelor’s Degree in Computer Science, Engineering, Electronics, Mathematics, or an equivalent highly technical field is required.
  • A minimum of 6 years of software engineering experience with a proven track record of successfully deploying AI models on devices is necessary.
  • Experience with GPU/TPU development and debugging is required.
  • Proficiency in Python and C++, along with familiarity with ML frameworks such as TensorFlow or PyTorch, is essential.
  • Advanced knowledge and hands-on experience with Linux is required.
  • Excellent problem-solving skills and the ability to dive deep into system architecture are necessary.
  • Strong software design skills are required.
  • Comfort in communicating and interacting with scientists, engineers, and product managers is essential.

Benefits:

  • Competitive salary and a 401k plan with employer match are offered.
  • Discretionary paid time off is provided.
  • Paid parental leave is available for all employees.
  • Medical, Dental, and Vision plans are included.
  • Fitness programs are available to employees.
  • Emotional and Mental Wellness support is provided.
  • Learning and Development programs are offered to encourage professional growth.
  • Snacks are available in the offices for employees.