MLOps Engineer
BJAK
KubernetesRay
Machine Learning Engineer – Diffusion, Vision
BJAK
full-time
Posted on:
Location Type: Remote
Location: Remote • 🇻🇳 Vietnam
Visit company websiteJob Level
Mid-LevelSenior
Tech Stack
FluxPyTorch
About the role
- 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.
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.
- Prior work in style transfer, personalization, or multimodal generation (Nice to Have).
- Experience with safety alignment in generative vision (NSFW detection, adversarial testing) (Nice to Have).
- Open-source contributions in diffusion or generative vision projects (Nice to Have).
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
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard skills
diffusion modelsgenerative visionStable DiffusionSDXLFluxDreamBoothLoRAPyTorchGPU optimizationstyle transfer
Soft skills
collaborationcommunicationproblem-solvingadaptabilityattention to detail