Axon is looking for a Machine Learning Engineer to contribute to developing AI solutions that transform the public safety space. You will enable new AI capabilities by supporting the training, evaluation, testing, and deployment of machine learning models on Axon devices and cloud.
What You'll Do
- Collaborate with scientists and product managers to build proof-of-concepts contributing to shaping the Axon of tomorrow.
- Architect and develop secure, privacy-preserving solutions to enable the continuous improvement of existing AI models.
- Architect platforms that accelerate research and AI product development.
- Collaborate with scientists in architecting and implementing state-of-the-art training techniques.
- Set high standards for ethical and responsible AI development.
What We're Looking For
- Bachelor’s Degree in Computer Science, Engineering, Electronics, Mathematics or an equivalent highly technical field.
- 6+ years of software engineering experience and a proven track record of successfully deploying AI models to the cloud.
- Experience with Infrastructure-as-code and cloud architecture.
- Proficiency in Python and C++, familiarity with ML frameworks such as TensorFlow or PyTorch.
- Advanced knowledge and hands-on experience with Linux.
- Excellent problem solving skills and ability to dive deep into system architecture.
- Excellent software design skills.
- Comfort communicating and interacting with scientists, engineers and product managers.
Nice to Have
- Master’s Degree/PhD in Computer Science, Engineering, Electronics, Mathematics or an equivalent highly technical field.
- Experience with LLMOps - evaluation, monitoring, quantization, teacher-learner, etc.).
- Hands-on experience with fine-tuning large parameter models.
- Familiarity with model encryption and privacy preserving AI techniques.
Technical Stack
- Python
- C++
- TensorFlow
- PyTorch
- Linux
Team & Environment
You will join a multidisciplinary team and report to the Manager, Machine Learning Engineering. The culture is mission-driven, focused on protecting life, connecting with candor and care, and tackling fast-paced, challenging work.
Benefits & Compensation
- Compensation range: USD 150,750 to USD 221,000
- 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
- Snacks in our offices
Work Mode
This is a hybrid position based in Seattle, Washington.
We are an equal opportunity employer that promotes justice, advances equity, values diversity and fosters inclusion.
