ML Engineer

Axon develops devices and cloud software for safety and justice, creating an ecosystem of products to protect life.
Vietnam
Machine Learning
Senior Software Engineer
In-Person
6+ years of experience
AI · Enterprise SaaS

Description For ML Engineer

Axon is a mission-driven company focused on protecting life through innovative safety and justice solutions. As an ML Engineer at Axon, you'll be at the forefront of developing AI solutions that transform public safety. The role involves working with a multidisciplinary team on various AI domains including computer vision, speech recognition, and natural language understanding.

You'll be contributing to products like Fleet, Axon Body, Axon Air, and VR, focusing on training, evaluation, testing, and deployment of machine learning models both on devices and in the cloud. The position requires strong technical expertise in AI/ML, with emphasis on model optimization and edge deployment.

The ideal candidate brings 6+ years of software engineering experience, strong AI model deployment background, and expertise in Python, C++, and ML frameworks. You'll be working in Vietnam, collaborating with global teams to build proof-of-concepts and architect privacy-preserving AI solutions.

Axon offers a competitive benefits package including medical insurance, flexible hours, and a modern work environment. The company values diversity and encourages applications from candidates who might not meet every requirement but are passionate about the mission.

This role provides an opportunity to make a real impact on public safety while working with cutting-edge AI technologies. You'll be part of a team that values candor, care, and continuous growth, all while contributing to meaningful solutions that protect life.

Last updated 7 days ago

Responsibilities For ML Engineer

  • Collaborate with scientists and product managers to build proof-of-concepts (POCs)
  • Architect and develop secure, privacy-preserving solutions for AI model improvement
  • Architect next generation continuous deployment framework for AI on device
  • Enable new AI features by optimizing models for the edge
  • Collaborate with scientists in implementing state-of-the-art training techniques

Requirements For ML Engineer

Python
Linux
TypeScript
  • Bachelor's Degree in Computer Science, Engineering, Electronics, Mathematics or equivalent
  • 6+ years of software engineering experience with AI model deployment
  • Experience with GPU/TPU development and debugging
  • Proficiency in Python and C++, familiarity with TensorFlow or PyTorch
  • Advanced knowledge and hands-on experience with Linux
  • Excellent problem solving and software design skills
  • Strong communication skills

Benefits For ML Engineer

Medical Insurance
  • Latest technologies office environment
  • Competitive salary
  • Medical insurance plan
  • Fun team activities and outings
  • Flexible working hours

Interested in this job?

Jobs Related To Axon ML Engineer

ML Engineer

Senior ML Engineer position at Axon, developing AI solutions for public safety products, requiring 6+ years of experience in software engineering and AI model deployment.

Machine Learning Engineer

Senior ML Engineer role at Axon focusing on developing AI solutions for public safety, including speech recognition, computer vision, and NLU models.

Machine Learning Engineer

Senior ML Engineer role at Axon focusing on developing AI solutions for public safety, including speech recognition, computer vision, and NLU models.

Sr. Research Scientist

Senior Research Scientist position at Axon, focusing on AI/ML research and development for public safety applications, offering competitive compensation and remote work options.

RF Machine Learning Engineer

Senior RF Machine Learning Engineer position at Axon, combining DSP and ML expertise to develop innovative wireless communication solutions in a hybrid work environment.