Hybrid ML Engineer - Diffusion & Vision (Remote-Friendly)

Bjak; Petaling Jaya; Hybrid; MYR 60,000 - 90,000; Full time

Job summary

A leading AI company in Malaysia seeks a Machine Learning Engineer to develop cutting-edge generative vision features. You'll customize diffusion models and build large-scale datasets while collaborating closely with product and design teams. Strong experience in diffusion models and proficiency in PyTorch are essential. This role offers a competitive salary, flexible working arrangements, and opportunities for accelerated learning.

Benefits

Competitive salary

Performance-based bonuses

Flexible work arrangement

High-impact role

Flat organizational structure

Cross-border exposure

Qualifications

• Strong experience with diffusion models and generative vision.
• Hands-on skills with DreamBooth and fine-tuning methods.
• Proficiency with PyTorch and dataset preparation.

Responsibilities

• Fine-tune and customize diffusion models using various methods.
• Build and annotate large-scale image datasets for generation.
• Develop pipelines to measure output fidelity and safety.
• Optimize performance through GPU memory and training techniques.
• Deploy features into production and monitor performance.

Skills

Diffusion models

Generative vision

PyTorch

Dataset preparation

GPU optimization

Job description

A leading AI company in Malaysia seeks a Machine Learning Engineer to develop cutting-edge generative vision features. You'll customize diffusion models and build large-scale datasets while collaborating closely with product and design teams. Strong experience in diffusion models and proficiency in PyTorch are essential. This role offers a competitive salary, flexible working arrangements, and opportunities for accelerated learning.