머신러닝 엔지니어 – 확산 모델/비전 / Machine Learning Engineer – Diffusion/Vision
BJAK
FluxPyTorch
机器学习工程师 – 扩散模型/视觉方向 Machine Learning Engineer – Diffusion/Vision
BJAK
full-time
Posted on:
Location Type: Hybrid
Location: 🇨🇳 China
Visit company websiteJob Level
Mid-LevelSenior
Tech Stack
FluxPyTorch
About the role
- 微调与定制 —— 使用 LoRA、DreamBooth 等参数高效方法训练并优化扩散模型（包括 SDXL、Flux、Stable Diffusion 等变体）
- 数据集构建 —— 构建、清洗和标注大规模图像数据集，涵盖自动生成描述、图像打标签、NSFW 过滤等
- 评估与对齐 —— 设计评估流程，衡量生成图像在保真度、多样性、风格一致性与安全性等维度的表现
- 性能优化 —— 应用 GPU 显存优化、潜空间扩散技巧与分布式训练等手段，实现高效扩展
- 部署与监控 —— 将扩散模型驱动的功能部署上线，并持续监控模型漂移、延迟与输出质量
- 跨职能合作 —— 与产品与设计团队协作，将视觉生成能力无缝融入用户体验中
Requirements
- 在扩散模型与视觉生成领域具备扎实经验（如 Stable Diffusion、SDXL、Flux 等）
- 熟练掌握 DreamBooth、LoRA/QLoRA 等微调技术
- 精通 PyTorch（优先）
- 具备图像数据准备能力，包括自动描述生成、标签标注、过滤与增强等
- 熟悉 GPU 优化、潜变量扩散技巧与高效训练方法
- 具备扎实的软件工程基础，包括算法、数据结构与整洁代码规范
Benefits
- 扁平化结构与真实的项目主导权
- 全面参与产品方向与共识式决策
- 弹性办公制度
- 高度影响力岗位，与产品、数据与工程团队深度协作
- 顶尖市场薪酬与绩效奖金
- 全球化视野的产品开发机会
- 丰厚福利 —— 住房租赁补贴、高质量公司餐厅、加班晚餐补助
- 健康、牙科与视力保险
- 全球旅行保险（适用于你和家属）
- 无限制、灵活的带薪休假制度
ATS Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard skills
LoRADreamBoothSDXLFluxStable DiffusionPyTorchGPU optimizationdata preparationalgorithmdata structure
Soft skills
cross-functional collaboration
Similar jobs on JobTailor
Senior Software Engineer, Perception Applications
Latitude
CloudPythonPyTorchTensorflow
Machine Learning Engineer, Autonomy
Blue River Technology
CloudPythonPyTorchTensorflow
Senior Algorithm Engineer, Image Processing
KLA
LinuxPythonPyTorchTensorflow
Senior Computer Vision System Performance Engineer
NVIDIA
CloudGRPCPythonPyTorch