Associate Machine Learning Engineer

The role

You will develop, train, and improve machine learning models for real-world applications. You will work with deep learning architectures, build ML pipelines, and deploy models into production environments.

What you will do

• Develop, train, evaluate, and improve machine learning models
• Work with deep learning architectures and modern ML frameworks
• Implement and experiment with neural networks, transformers, and generative models
• Perform data preprocessing, feature engineering, model evaluation, and error analysis
• Experiment with model architectures, training configurations, and hyperparameters
• Explore modern Generative AI technologies including diffusion models, transformers, and LLMs
• Work with pretrained models and adapt them for product requirements
• Build and maintain ML training and inference pipelines
• Optimize inference latency, GPU utilization, and memory consumption
• Collaborate with backend engineers to integrate ML models into production systems

Required qualifications

• Bachelor's degree in Computer Science, Artificial Intelligence, Data Science, or a related field
• 0–2 years of experience in Machine Learning, Deep Learning, or relevant practical projects
• Strong programming skills in Python
• Good understanding of fundamental Machine Learning concepts, including supervised and unsupervised learning, model evaluation, and optimization
• Hands-on experience with PyTorch or TensorFlow
• Understanding of neural networks, backpropagation, gradient descent, and common deep learning architectures
• Familiarity with NumPy, Pandas, and Scikit-learn
• Understanding of CNNs, Transformers, or other deep learning architectures
• Experience implementing ML projects beyond basic tutorials or coursework

Nice to have

• Familiarity with Git, Linux, and basic software development practices
• Exposure to Generative AI, LLMs, diffusion models, or computer vision
• Familiarity with Hugging Face, pretrained models, and model fine-tuning
• Understanding of CUDA, GPU computing, or model inference optimization
• Experience reading and implementing research papers

About the company

ImagineArt is a fast-growing GenAI company that has built its own image generation model ranked #3 globally for photorealism.