Senior Machine Learning Engineer II at Axon

Axon

Seattle, Washington, United States·1d ago

SeniorFull-timePythonC++AWSMachine LearningTensorFlowPyTorchAIML

Salary Range

$132,600 - $179,400 /year

Estimated

This salary is estimated based on similar roles. The actual salary may vary.

About the Role

We are looking for a ML Engineer to join our team at Axon. This role focuses on architecting and implementing the platform used by Axon scientists to bring safe, secure, reliable AI features to Axon devices. You will collaborate closely with scientists and device engineers to enable new AI capabilities for Axon products.

Key Responsibilities

→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.
→Introduce new techniques leading to tangible impact in terms of model fairness, performance, and platform scalability.

Requirements

✓Bachelor’s Degree in Computer Science, Engineering, Physics, Mathematics or an equivalent highly technical field.
✓10+ 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.