Machine Learning Engineer

Overview Join to apply for the Machine Learning Engineer (Diffusion / Vision) role at BJAK . This role is a global position with remote work arrangements. You’ll work with regional teams across product, engineering, operations, infrastructure and data to build and scale impactful AI solutions. Why This Role Matters You’ll fine-tune state-of-the-art models, design evaluation frameworks, and bring AI features into production. Your work will help ensure our models are intelligent, safe, trustworthy, and impactful at scale. 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. What It’s Like Ownership and independence; action-oriented with prototypes and iterations. Stay calm and effective in startup environments, shifting priorities and building from zero. Bias for speed: deliver valuable results now rather than waiting for a perfect version later. Embrace feedback and failure as part of growth; lift others as you go. Humility, hunger, hustle, and collaboration. Requirements Strong experience with diffusion models and generative vision (Stable Diffusion, SDXL, Flux, etc.). Hands-on skills with DreamBooth, LoRA/QLoRA, and fine-tuning methods. Proficiency with PyTorch (preferred). Experience in dataset preparation (captioning, tagging, filtering, augmentation). Knowledge of GPU optimization, latent diffusion, and efficient training techniques. Strong foundations in software engineering, algorithms, and clean code practices. Nice to Have Prior work in style transfer, personalization, or multimodal generation. Experience with safety alignment in generative vision (NSFW detection, adversarial testing). Open-source contributions in diffusion or generative vision projects. What You’ll Get Flat structure and real ownership Full involvement in direction and consensus decision making Flexibility in work arrangement High-impact role with visibility across product, data, and engineering Top-of-market compensation and performance-based bonuses Global exposure to product development Perks including housing rental subsidies, company cafeteria, and meals Health, dental & vision insurance Global travel insurance for you and dependents Unlimited, flexible time off Our Team & Culture We’re a dense, high-performance team focused on high-quality work and global impact. We behave like owners and value speed, clarity, and relentless ownership. If you’re hungry to grow and care deeply about excellence, join us. About BJAK BJAK is Southeast Asia’s #1 insurance aggregator with 8M+ users. Fully owned by its employees, headquarter 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 Entry level Employment type Full-time Job function Engineering and Information Technology Industries: Software Development Referrals increase your chances of interviewing at BJAK. Get notified about new Machine Learning Engineer jobs in Singapore, Singapore. #J-18808-Ljbffr