Senior Machine Learning Engineer

Who you are

You have 5+ years of experience in machine learning engineering, with a strong background in developing and deploying machine learning models. Your expertise in Python is complemented by your knowledge of frameworks such as TensorFlow and PyTorch, allowing you to build robust AI solutions. You are familiar with various AI domains, including computer vision and natural language processing, and have a passion for applying technology to solve real-world problems. You thrive in collaborative environments, working closely with cross-functional teams to drive innovation and impact. You are committed to continuous learning and growth, eager to stay updated with the latest advancements in AI and machine learning. You understand the importance of ethical AI practices and are dedicated to contributing to projects that promote safety and justice.

Desirable

Experience with cloud platforms such as AWS or GCP is a plus, as is familiarity with MLOps practices. A background in public safety or related fields can enhance your contributions to the team. You are open to exploring new technologies and methodologies, always looking for ways to improve processes and outcomes.

What you'll do

As a Senior Machine Learning Engineer at Axon, you will play a crucial role in developing AI capabilities that transform public safety solutions. You will collaborate with scientists and product managers to create proof-of-concepts that shape the future of Axon's products. Your responsibilities will include training, evaluating, and deploying machine learning models on Axon devices and cloud infrastructure. You will work on diverse AI applications, from computer vision to speech recognition, ensuring that the solutions you develop are effective and reliable. You will also contribute to the architectural design of machine learning systems, ensuring they are scalable and maintainable. Your work will directly impact the mission of Axon, helping to accelerate justice and protect lives through innovative technology.

What we offer

At Axon, you will be part of a mission-driven team that values collaboration and diverse perspectives. We offer a hybrid work environment, allowing you to balance in-office and remote work. You will have opportunities for professional development and growth, with access to resources that support your career advancement. Our culture emphasizes the importance of work-life balance and employee well-being, ensuring that you can thrive both personally and professionally. Join us in making a difference in the world through technology that matters.