Machine Learning Engineer (Diffusion / Vision)

Overview

Join to apply for the Machine Learning Engineer (Diffusion / Vision) role at BJAK .

Transform Visual Models into Real-World Applications. 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.

This role is based in Malaysia with a hybrid work arrangement - combining flexible remote work with in-office collaboration at our HQ. You’ll work closely with regional teams across product, engineering, and design to bring cutting-edge generative vision features to market.

Why This Role Matters

We are scaling generative vision capabilities that reshape how people experience financial and consumer products. As a Machine Learning Engineer (Diffusion / Vision), you’ll fine-tune diffusion models, design evaluation frameworks, and deploy AI features into production. Your work ensures our image systems are creative, safe, and optimized for real-world impact.

What You’ll Do

• Fine-tune & Adapt – Train and customize diffusion models (SDXL, Flux, Stable Diffusion variants) using LoRA, DreamBooth, and other parameter-efficient methods.
• Curate Datasets – Build, clean, and annotate large-scale image datasets with captioning, tagging, and NSFW filtering for safe and aligned generation.
• Evaluate & Align – Develop pipelines to measure fidelity, diversity, style adherence, and safety across generated outputs.
• Optimize Performance – Apply GPU memory optimization, latent diffusion tricks, and distributed training for efficient scaling.
• Deploy & Monitor – Ship diffusion-powered features into production with monitoring for drift, latency, and quality.
• Collaborate & Deliver – Work with product and design to integrate generative vision capabilities into user experiences.

You’ll Thrive Here If You…

Requirements

Nice to Have

What You’ll Get

About BJAK

BJAK is Southeast Asia’s #1 insurance aggregator with 8M+ users, fully owned by its employees. Headquartered in Malaysia and operating in Thailand, Taiwan, and Japan, we help millions of users access transparent and affordable financial protection through Bjak.com. We simplify complex financial products through cutting-edge technologies, including APIs, automation, and AI, to build the next generation of intelligent financial systems.

If you're excited to build real-world AI systems and grow fast in a high-impact environment, we’d love to hear from you.

Seniority level

Employment type

Job function

Industries

#J-18808-Ljbffr