Machine Learning Engineer (Diffusion / Vision) - Remote

Overview

At Bjak, we’re building AI systems that reach millions across Southeast Asia. Our Machine Learning team is pushing the frontier of image generation, personalization, and safe deployment of diffusion models. From powering creative assets to enabling user-facing visual tools, we make diffusion models fast, scalable, and aligned with our product vision.

In Short

• Fine-tune and adapt diffusion models using LoRA, DreamBooth, and other methods.
• Build, clean, and annotate large-scale image datasets.
• Develop pipelines to measure fidelity, diversity, and safety of outputs.
• Apply GPU memory optimization and distributed training.
• Ship diffusion-powered features into production.
• Collaborate with product and design teams.

Requirements

• Experience with diffusion models and generative vision.
• Hands-on skills with fine-tuning methods.
• Proficiency with PyTorch.
• Experience in dataset preparation.
• Knowledge of GPU optimization techniques.
• Strong foundations in software engineering and clean code practices.

Benefits

• Competitive salary and performance-based bonuses.
• Flexible work arrangement.
• High-impact role with visibility across teams.
• Flat structure where ideas are heard.
• Accelerated learning and growth opportunities.
• Cross-border exposure to regional product development.