小米｜AI 模型加速优化算法工程师

岗位职责 1、负责大模型、多模态大模型、AIGC、自动驾驶、多媒体/ISP 等 AI 模型端侧加速优化算法的研发与探索； 2、负责模型训练调优及压缩优化，包括但不限于量化、剪枝、蒸馏、稀疏化、小型化、NAS、推理加速等技术研发； 3、与芯片团队协同开展算法-芯片联合设计与优化，提供面向自研芯片的模型加速解决方案； 4、支持业务模型在自研芯片上的训练优化、部署交付及性能持续优化。

岗位要求 1、在大模型、AIGC、自动驾驶、多媒体/ISP、AI 芯片等方向上具有相关项目或实习经验； 2、熟悉模型量化、剪枝、蒸馏、稀疏化、训练调优及性能优化等模型压缩与加速技术； 3、熟练使用 PyTorch，熟悉 Transformer、Diffusion 等模型，掌握 TensorRT、ONNX Runtime、TVM 等至少一种部署或优化框架； 4、有芯片算法联合优化、端侧 AI 部署或模型加速项目经验者优先； 5、有大模型/AIGC/自动驾驶/多媒体 ISP 模型优化经验，或相关论文、专利、开源项目者优先。