Senior Machine Learning Engineer
Axon is seeking a Senior Machine Learning Engineer to join our team in San Francisco, California. In this role, you will contribute to developing AI solutions that transform the public safety space, supporting our mission to accelerate justice, protect truth, and save lives. You will work closely with other machine learning engineers and scientists to enable new AI capabilities for Axon's products, including Fleet, Draft One, Axon Air, and VR. This position offers exposure to a wide range of AI domains and applications, from computer vision to speech recognition and natural language understanding.
As a Senior Machine Learning Engineer, your key responsibilities will include collaborating with scientists and product managers to build proof-of-concepts that shape the future of Axon. You will architect and develop secure, privacy-preserving solutions to enable the continuous improvement of existing AI models. Additionally, you will design platforms that accelerate research and AI product development, and implement state-of-the-art training techniques in collaboration with scientists. Maintaining high standards for ethical and responsible AI development is a critical aspect of this role.
The ideal candidate will possess a Bachelor's Degree in Computer Science, Engineering, Electronics, Mathematics, or a related technical field, along with at least 6 years of software engineering experience and a proven track record of successfully deploying AI models to the cloud. Proficiency in Python and C++, familiarity with machine learning frameworks such as TensorFlow or PyTorch, and advanced knowledge of Linux are essential. Strong problem-solving skills, excellent software design abilities, and effective communication with scientists, engineers, and product managers are also required.
Axon offers a competitive salary and 401k with employer match, discretionary paid time off, paid parental leave for all employees, and comprehensive medical, dental, and vision plans. Additional benefits include fitness programs, emotional and mental wellness support, learning and development programs, and in-office snacks. Please note that benefits may vary depending on the nature of your employment and the location where you work.
At Axon, we foster a fast-paced, challenging, and meaningful work environment where employees take ownership and drive real change. We value diverse perspectives and encourage collaboration with candor and care. Joining our team means contributing to a mission that matters at a company where you matter.