BJAK is looking for a Machine Learning Engineer specializing in Diffusion and Vision models to bring generative AI capabilities into production. You will fine-tune state-of-the-art models, design evaluation frameworks, and ensure they are safe, trustworthy, and impactful at scale.
What You'll Do
- Train and customize diffusion models (SDXL, Flux, Stable Diffusion variants) using LoRA, DreamBooth, and parameter-efficient methods.
- Build, clean, and annotate large-scale image datasets with captioning, tagging, and NSFW filtering for safe and aligned generation.
- Develop pipelines to measure fidelity, diversity, style adherence, and safety across generated outputs.
- Apply GPU memory optimization, latent diffusion tricks, and distributed training for efficient scaling.
- Ship diffusion-powered features into production with monitoring for drift, latency, and quality.
- Work with product and design to integrate generative vision capabilities into user experiences.
What We're Looking For
- Strong experience with diffusion models and generative vision (Stable Diffusion, SDXL, Flux).
- Hands-on skills with DreamBooth, LoRA/QLoRA, and fine-tuning methods.
- Proficiency with PyTorch.
- 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.
Technical Stack
- PyTorch
- Stable Diffusion
- SDXL
- Flux
- LoRA
- QLoRA
- DreamBooth
Team & Environment
Flat structure & real ownership. You will work closely with regional teams across product, engineering, operations, infrastructure and data.
Benefits & Compensation
- 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
- Housing rental subsidies
- Quality company cafeteria
- Overtime meals
- Health, dental & vision insurance
- Global travel insurance (for you & your dependents)
- Unlimited, flexible time off
Work Mode
This is a remote position open to candidates globally.
BJAK is an equal opportunity employer.
