Axon is looking to develop AI solutions that transform the public safety space and support its mission to accelerate justice, protect truth, and save lives by enabling new AI capabilities for its products.
Requirements
- 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.
- 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.
Responsibilities
- Collaborate with scientists and product managers to build proof-of-concepts (POCs) 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.
- Support the training, evaluation, testing and deployment of machine learning models on Axon devices and cloud.
Other
- Bachelor’s Degree in Computer Science, Engineering, Electronics, Mathematics or an equivalent highly technical field.
- Master’s Degree/PhD in Computer Science, Engineering, Electronics, Mathematics or an equivalent highly technical field.
- Comfort communicating and interacting with scientists, engineers and product managers.
- Location: Remotely from US or Hybrid from our office in Seattle, Washington.
- Reports to: Manager, Machine Learning Engineering
