跳转到主要内容

生成式AI技术专家

Technology
小米科技
北京市, 中国2个月前
全职

职位描述

职位描述

1、负责生成式AI在手机影像场景的算法研发与落地,包括但不限于 AI 消除、AI 扩图、超分重建、画质增强、人像/风格化生成等能力。

2、跟进扩散模型(Diffusion / Flow Matching)、多模态大模型等前沿方向,完成技术选型、原型验证到产品化的全流程。

3、面向端侧算力约束做模型压缩与推理优化(蒸馏、量化、剪枝、少步采样),与芯片及平台团队协同完成 NPU/GPU 部署。

4、参与影像数据链路建设,包括训练数据构建、评测体系与主客观画质对齐。

5、与产品、交互、系统团队协作定义功能形态,推动能力在旗舰机型上量产交付。

职位要求

1、硕士及以上学历,计算机视觉、机器学习、图像处理等相关专业。

2、熟练掌握 PyTorch,具备扎实的深度学习工程能力与代码功底。

3、深入理解扩散模型原理与主流框架(Stable Diffusion / SDXL / DiT / ControlNet / LoRA 等),有独立训练或微调大规模生成模型的经验。

4、具备至少一项完整落地经验:图像生成、图像编辑、超分、去噪、ISP 相关算法或视频生成。

5、良好的沟通与协作意识,能在跨团队环境中推动技术方案落地。

Keywords
PyTorch

对这个职位感兴趣吗?