BJAK is hiring a Machine Learning Engineer focused on Diffusion and Vision models. You will fine-tune diffusion models, design evaluation frameworks, and deploy AI features into production to scale generative vision capabilities that reshape how people experience financial and consumer products. Your work will ensure image systems are creative, safe, and optimized for real-world impact.

What You'll Do

Train and customize diffusion models (SDXL, Flux, Stable Diffusion variants) using LoRA, DreamBooth, and other 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, etc.).
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

You will join a flat structure, lean high-performance team. Treat every challenge as your own, regardless of title. Enjoy switching between researcher, engineer, and product thinker. We prototype fast, test, iterate, and improve without waiting for perfect conditions. Stay calm and effective in startup chaos. We have a bias for speed, believing something valuable now beats perfect later. Learn fast from feedback and failure. Care deeply about both user experience and technical excellence. Show up with humility, hunger, and hustle. In our flat structure, your work is seen and your ideas are heard. We value speed, clarity, and relentless execution.