Senior Machine Learning Engineer II
Axon
24 days ago
Seattle, WA, USA
Senior / Staff+
H1B Sponsor
Base Salary
$157k - $251k/yr
Responsibilities
- Architect and develop secure, privacy-preserving, on-device solutions for AI models.
- Collaborate with scientists to implement edge distributed training techniques.
- Implement on-device monitoring solutions for continuous model improvement.
- Develop innovative model compression solutions for AI at the edge.
- Introduce new techniques to enhance model fairness, performance, and scalability.
Requirements
- Bachelor’s Degree in Computer Science, Engineering, Physics, Mathematics, or a related field.
- 10+ years of software engineering experience with large-scale distributed platforms.
- Experience with AI on chips and on-device model deployment.
- Proficiency in Python and C++, with familiarity in ML frameworks like TensorFlow or PyTorch.
- Advanced knowledge of on-chip development and excellent problem-solving skills.
- Strong communication skills to interact with scientists, engineers, and product managers.
Benefits
- Competitive salary and 401k with employer match.
- Discretionary paid time off and paid parental leave.
- Medical, Dental, and Vision plans.
- Fitness programs and emotional & mental wellness support.
- Learning & development programs and employee resource groups.
Tech Stack
C++PythonPyTorchTensorFlow
Categories
AI & MLData Science