Machine Learning Engineer II

Summary

Join Axon and contribute to building AI-powered products that enhance public safety. As a Machine Learning Engineer, you will architect and implement platforms for training and deploying machine learning models, collaborating with scientists and engineers. You will optimize model training and evaluation, develop innovative distributed solutions, and contribute to the advancement of privacy-preserving AI. The ideal candidate possesses extensive software engineering experience, proficiency in Python and ML frameworks, and expertise in cloud environments. Axon offers a competitive salary and benefits package, including medical insurance, flexible working hours, and professional development opportunities.

Requirements

• Bachelor’s Degree in Computer Science, Engineering, Physics, Mathematics or an equivalent highly technical field
• 8+ years of software engineering experience and a proven track record of successfully training
• Proficiency in python, familiarity with ML frameworks such as TensorFlow, Keras, PyTorch
• Advanced knowledge and hands-on experience with at least one cloud environment (such as AWS, Microsoft Azure Oracle, or Google Cloud) and Infrastructure as Code for deploying Platforms at scale
• Experience with CI/CD solutions in the context of MLOps including automation with IaC (e.g., using terraform)
• Excellent problem solving and software design skills
• Comfort communicating and interacting with scientists, engineers and ML/product managers

Responsibilities

• Architect solutions to train and evaluate models on a distributed architecture
• Optimize the evaluation and training of models within Axon
• Architect novel, strategic distributed solutions for continuous model improvement
• Deliver novel, strategic solutions to accelerate innovation
• Influence the AI community by developing state-of-the-art solutions for privacy-preserving distributed model training

Preferred Qualifications

• Master’s Degree/PhD in Computer Science, Engineering, Electronics, Mathematics or an equivalent highly technical field
• Hands-on experience in solving Computer Vision or Speech Recognition problems in a business setting
• Proven track record of significant contribution to distributed architectures for model training
• Familiarity with responsible AI, de-biasing, encryption and de-identifying techniques

Benefits

• An awesome office/working environment with the latest technologies
• Competitive salary and benefits including a great medical insurance plan for you and your family
• Free lunch, gym, phone stipend, parking
• Fun team activities and outings
• Flexible working hours
• Opportunities for training and rotations in the US
• Opportunities to ride along with real US police officers in real life situations, see them use technology, and get inspired