Machine Learning Engineer II
HybridHo Chi Minh City, Ho Chi Minh City (HCMC), Vietnam
Job Summary
Architect and develop secure, privacy-preserving AI solutions for Axon's public safety devices and cloud infrastructure. Collaborate with scientists and product managers to build proof-of-concepts and optimize models for edge deployment across Fleet, Axon Body, and Axon Air products. Architect the next generation of continuous deployment frameworks for AI on device and cloud while implementing state-of-the-art training techniques. This role is based in Ho Chi Minh City with a hybrid schedule (Tuesdays through Fridays onsite). Requires 6+ years of software engineering experience and proficiency in Python, C++, and ML frameworks like TensorFlow or Pytorch.
Required Qualifications
- Bachelor's Degree in Computer Science, Engineering, Electronics, Mathematics or an equivalent highly technical field
- 6+ years of software engineering experience and a proven track record of successfully deploying AI models on devices
- Experience with GPU/TPU development and debugging
- Proficiency in Python and C++, familiarity with ML frameworks such as TensorFlow, or PyTorch
- Advanced knowledge and hands-on experience with Linux
- Excellent problem solving skills and ability to dive deep into system architecture
- Excellent software design skills
- Comfort communicating and interacting with scientists, engineers and product managers
- Work Location: This role is based out of our Ho Chi Minh City office and follows a hybrid schedule
- We rely on in-person collaboration and ask that team members work onsite Tuesdays through Fridays, with the flexibility to work remotely on Mondays, unless there is an approved workplace accommodation
Desired Qualifications
- Master's Degree/PhD in Computer Science, Engineering, Electronics, Mathematics or an equivalent highly technical field
- Experience with model compression (quantization, model distillation, ...)
- Hands-on experience in solving Computer Vision or Natural Language Understanding problems in a business setting
- Familiarity with model encryption and privacy preserving AI techniques
Hiring someone like this?
Get your role in front of qualified candidates on Sorce.