Remote Senior Machine Learning Engineer (Remote)

Description:

• As a Senior Machine Learning Engineer at Axon, you will play a crucial role in architecting and implementing the platform used by Axon scientists to enhance public safety.
• You will collaborate with other engineers to enable new AI capabilities for Axon products like Fleet, Axon Body, and Axon Air by supporting the training, evaluation, testing, and deployment of machine learning models.
• The role involves exposure to various domains and applications such as computer vision, speech recognition, and NLU, LLMs.
• The ideal candidate will have a track record of managing large-scale distributed platforms for secure and compliant research in the cloud, understanding the AI lifecycle, and supporting scientists in developing state-of-the-art AI solutions for transforming public safety.

Requirements:

• Bachelor’s Degree in Computer Science, Engineering, Physics, Mathematics, or a related technical field.
• 8+ years of software engineering experience with a focus on training.
• Proficiency in Python and familiarity with ML frameworks like TensorFlow, Keras, PyTorch.
• Advanced knowledge and hands-on experience with at least one cloud environment (AWS, Microsoft Azure, Oracle, or Google Cloud) and Infrastructure as Code for deploying platforms at scale.
• Experience with CI/CD solutions in MLOps context, including automation with IaC (e.g., using Terraform).
• Strong problem-solving and software design skills.
• Ability to effectively communicate and collaborate with scientists, engineers, and ML/product managers.

Benefits:

• Access to an excellent office/working environment equipped with the latest technologies.
• Competitive salary and benefits package, including comprehensive medical insurance for you and your family.
• Additional perks like free lunch, gym access, phone stipend, parking, and team activities.
• Flexible working hours and opportunities for training and rotations in the US.
• Unique opportunities to accompany real US police officers in real-life situations, witness technology use, and gain inspiration.