Associate ML Engineer - Generative AI & Production

ImagineArt; Islamabad; On-site; PKR 900,000 - 1,500,000; Full time

Job summary

ImagineArt is hiring an Associate Machine Learning Engineer to design, train, and deploy ML models powering our GenAI products. You’ll work with PyTorch, Transformers, and diffusion-based approaches to build scalable pipelines and production-ready systems.

The role blends research with hands-on implementation on an on-site team in Islamabad. You’ll collaborate with ML researchers and engineers to optimize inference, reduce latency, and push state-of-the-art AI in real-world creative

Qualifications

• 0–2 years of hands-on ML or DL project experience.
• Strong Python programming skills.
• Hands-on with PyTorch or TensorFlow.
• Solid understanding of ML basics: supervision, unsupervised learning, evaluation, and optimization.
• Familiarity with NumPy, Pandas, and Scikit-learn.
• Familiarity with CNNs, Transformers, and other modern architectures.
• Experience implementing ML projects beyond tutorials.
• Good knowledge of Git, Linux, and software development basics.
• Exposure to Generative AI, LLMs, diffusion models or computer vision.
• Knowledge of Hugging Face and model fine-tuning.
• Understanding CUDA, GPU computing, or inference optimization.

Responsibilities

• Develop, train, evaluate, and improve ML models for real-world applications.
• Work with deep learning architectures and ML frameworks.
• Assist in implementing neural networks, transformers, and generative models.
• Perform data preprocessing, feature engineering, model evaluation, and error analysis.
• Experiment with different model architectures, training configurations, and hyperparameters.
• Collaborate with backend engineers to integrate ML models into production systems.
• Write clean, modular, and maintainable Python code.
• Deploy models into production environments.
• Optimize inference latency, GPU utilization, and memory consumption.

Skills

Python

PyTorch

TensorFlow

NumPy

Pandas

Scikit-learn

Git

Linux

CNNs

Transformers

Generative AI

LLMs

Diffusion models

Hugging Face

CUDA

GPU optimization

Education

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

Tools

Hugging Face

Git

Job description

ImagineArt is hiring an Associate Machine Learning Engineer to design, train, and deploy ML models powering our GenAI products. You’ll work with PyTorch, Transformers, and diffusion-based approaches to build scalable pipelines and production-ready systems.

The role blends research with hands-on implementation on an on-site team in Islamabad. You’ll collaborate with ML researchers and engineers to optimize inference, reduce latency, and push state-of-the-art AI in real-world creative