Software Engineer (AI/ML)

Emage Innovation Sdn Bhd; Iskandar Puteri; On-site; MYR 60,000 - 110,000; Full time

Job summary

Emage Innovation Sdn Bhd in Nusajaya, Johor, is seeking a talented Machine Learning Engineer to design, train, and deploy computer vision models. You will apply PyTorch/TensorFlow, develop CNNs and ViTs, and collaborate with engineering teams to integrate AI into desktop, embedded, or cloud applications.

You should have a degree in CS/AI and at least 1 year in ML/DL/CV, with strong Python skills and OpenCV knowledge.

Qualifications

• Bachelor's or Master's in CS/AI or related field.
• Minimum 1 year hands-on ML/DL/CV experience.
• Strong Python programming.
• Experience with PyTorch, TensorFlow, or Keras.

Responsibilities

• Design, train, and deploy CV models using PyTorch/TensorFlow.
• Build scalable training pipelines for single/multi-GPU and distributed settings.
• Collaborate with software engineers to integrate AI into apps.
• Optimize models for high-performance inference with CUDA/TensorRT.
• Prepare and annotate datasets; evaluate model performance.

Skills

Python

PyTorch

TensorFlow

Keras

OpenCV

CUDA

cuDNN

ViT

CNNs

Linux

Teamwork

Education

Bachelor's or Master’s degree in Computer Science, Artificial Intelligence, Data Science, Electronics, or related field

Tools

Docker

Git

NVIDIA Jetson

HALCON

TensorRT

ONNX

CUDA

Job description

Jora Malaysia will close on 9th September 2026. Thank you for being with us, we are cheering you on as you continue your career journey.

Emage Innovation Sdn Bhd - Nusajaya, Johor

Design, develop, train, evaluate, and deploy Machine Learning and Deep Learning models using Python.

Develop AI solutions for object detection, image classification, image segmentation, anomaly detection, and industrial inspection.

Implement and optimize deep learning architectures using TensorFlow, PyTorch, or Keras.

Apply transfer learning, fine-tuning, and data augmentation techniques to improve model accuracy and robustness.

Design and implement advanced neural network architectures including CNNs, Vision Transformers (ViTs), RNNs, Autoencoders, GANs, and Transformer-based models.

Build scalable training pipelines supporting single-GPU, multi-GPU, and distributed training (DDP).

Optimize AI models for high-performance inference using CUDA, TensorRT, ONNX, and cuDNN.

Perform image preprocessing and enhancement using OpenCV or similar image processing libraries.

Prepare, clean, annotate, and manage datasets for model training and validation.

Evaluate model performance using appropriate metrics and continuously improve model accuracy and inference speed.

Collaborate with software engineers to integrate AI models into desktop, embedded, or cloud-based applications.

Participate in code reviews, technical discussions, and maintain technical documentation.

Required Skills

Bachelor's or Master's degree in Computer Science, Artificial Intelligence, Data Science, Electronics, or a related field.

Minimum 1 years of hands-on experiencein Machine Learning, Deep Learning, or Computer Vision.

Strong programming skills in Python.

Strong understanding of Machine Learning and Deep Learning concepts.

Hands-on experience with PyTorch, TensorFlow, or Keras.

Strong knowledge of Computer Vision using OpenCV.

Experience developing one or more of the following:

Image Classification

Anomaly Detection (PatchCore, Autoencoders, EfficientAD, or similar)

Good understanding of image processing techniques such as filtering, thresholding, morphology, feature extraction, and image enhancement.

Experience with GPU computing and model optimization using CUDA-enabled environments.

Familiarity with Linux and Windows development environments.

Strong analytical, debugging, and problem-solving skills.

Good communication and teamwork skills.

Preferred Skills (Good to Have)

Experience with NVIDIA TensorRT, ONNX Runtime, CUDA, and cuDNN.

Experience with distributed training using DDP or multi-GPU environments.

Familiarity with Docker and containerized AI deployments.

Experience with NVIDIA Jetson platforms or edge AI deployment.

Familiarity with Git and version control systems.

Exposure to HALCON or other industrial machine vision libraries.

Experience integrating AI models into production software applications.

Experience working on manufacturing, industrial automation, quality inspection, or machine vision projects.

Basic knowledge of cloud platforms such as AWS, Azure, or Google Cloud is an added advantage.

What We Offer

Opportunity to work on cutting-edge AI, Machine Learning, and Computer Vision technologies.

Exposure to industrial automation, machine vision, and smart manufacturing solutions.

Hands-on experience with high-performance GPU computing and AI optimization.

Opportunity to work with the latest Deep Learning and Generative AI models.

Collaborative engineering environment with continuous learning and career growth.

Opportunity to develop production-ready AI solutions deployed in real-world manufacturing environments.

Nice to Have

Candidates with experience in any of the following will have an added advantage: