AI Developer
CACI International Inc
AWSEC2FlaskLinuxPythonSQLUnix
Machine Learning Engineer, Diffusion/Vision
BJAK
full-time
Posted on:
Location: 🇺🇸 United States
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.
- What Is It Like – Ownership and independence; prototype, test, and iterate quickly; thrive in startup environment; bias for speed; growth mindset; humility and hustle.
- About BJAK – Southeast Asia’s #1 insurance aggregator with 8M+ users; HQ in Malaysia; operates in Thailand, Taiwan, and Japan.
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.
