Senior Staff Machine Learning Engineer

Job description

• Design, develop, and optimize advanced AI models (e.g., CNN, NLP, VLM, LLM) for deployment on device platforms such as Intel, NVIDIA, Ambarella, and Hailo.
• Perform model conversion and adaptation using frameworks like TensorFlow/TensorFlow Lite, PyTorch, ONNX, and TensorRT, ensuring compatibility and performance.
• Develop and fine-tune custom kernels and operators with low-level programming (C/C++, CUDA) to maximize efficiency.
• Collaborate on hardware-software co-design initiatives to integrate AI solutions seamlessly.
• Optimize model performance, latency, and power efficiency on device.
• Provide technical expertise, detailed documentation, and insights to support cross-functional teams.

#R54169

Your skills and experience

• Bachelor's/Master's degree in Computer Science, Machine Learning, or any related fields.
• 3- 5 years of hands-on experience in AI model development, optimization, and deployment.
• Expert in using framework TensorFlow, PyTorch, and conversion tools (ONNX, TensorRT, OpenVINO).
• Strong understanding of AI model architectures (CNN, NLP, VLM, LLM) and proficiency in training and evaluation techniques.
• Proven experience with hardware architectures and optimization techniques.
• Creative, proactive, and independence in learning and problem-solving.
• Outstanding analytical and problem-solving skills.
• Team player.
• Fluent in verbal and written in English language.

Why you'll love working here

• Compensation & bonus: 13th & 14th salary, AIP bonus, Holidays, Tet, and Long year service …
• Social insurance, Health insurance, Unemployment insurance: by Social Insurance and Labor Law
• The regime of annual leave, company trip, and checkup examination
• Award for marriage, newborn
• We have AON insurance package for employee, spouse, and children every year
• You will be trained, learned & work with the best technical managers who help you improve various dev skills & career path
• You’ll love working in our dynamic environment employees, young & active
• We love sport activities, as marathon, football, swimming,...
• Working time: From Monday to Friday | 08:30-12:00 & 13:00-17.30