Remote (Global)

Bjak is hiring a Machine Learning Engineer (Diffusion / Vision) (Remote)

Responsibilities

  • 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.

Benefits

  • Flat structure & 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
  • Lots of perks - housing rental subsidies, a quality company cafeteria, and overtime meals
  • Health, dental & vision insurance
  • Global travel insurance (for you & your dependents)
  • Unlimited, flexible time off

Work Arrangement

Remote (Worldwide)

Team

Structure: Flat structure & real ownership

Additional Information

  • Global travel insurance (for you & your dependents)
  • Unlimited, flexible time off
Required Skills
PytorchComputer VisionDeep LearningGenerative AI
About company
Bjak
Bjak is focused on providing access to affordable and sustainable financial services for people in ASEAN. Headquartered in Malaysia, Bjak is the largest insurance portal in Southeast Asia. Their main portal, Bjak.com, helps millions find the insurance policy with the best value and highest coverage. They invest in technology such as Custom API, trading systems, and data science to enable easy access to financial services.
All jobs at Bjak Visit website
Job Details
Category data
Posted 7 months ago