Machine Learning Engineer II

📋 Description

• Collaborate with scientists and PMs to build POCs shaping Axon’s future.
• Architect and develop secure, privacy-preserving AI solutions.
• Design next-gen CD framework for AI on device and cloud.
• Enable AI features on Axon devices by optimizing models for the edge.
• Collaborate on state-of-the-art training techniques with scientists.

🎯 Requirements

• Bachelor’s Degree in CS, Engineering, Electronics, Math, or equivalent.
• 6+ years software engineering and deploying AI models on devices.
• Experience with GPU/TPU development and debugging.
• Python and C++, with TensorFlow or PyTorch.
• Advanced Linux skills and system-level knowledge.
• Strong problem solving and system architecture skills.

🎁 Benefits

• Medical, Dental and Vision Insurance
• Robust Paid Time Off policy
• Bonuses
• Lunch allowance
• Cell phone stipend
• Free LinkedIn Learning account or Udemy account