高性能计算工程师 - 图像视频生成推理优化

地点

北京

团队

美图影像研究院

招聘类型

社招

尚未披露：雇佣形式、薪资、雇主申请截止时间

投递在官方渠道完成，完整外部回执留在本人工作区；已登录时，通过 CLI 明确记录的选择、本人自报投递和渠道观察三类最小事件默认上报平台。浏览或保存岗位不会触发上报，外部投递不会出现在“我的申请”里。

• 对Diffusion/DiT/VLM等图像和视频生成模型做生产推理优化，覆盖图算子/显存、PTQ/QAT/蒸馏、TensorRT/vLLM/SGLang后端、视频缓存、请求调度和性能评测，并推动算法方案生产落地。

任职要求与期待

• 计算机、电子、AI或高性能计算相关背景，本科及以上，能使用C++与Python。
• 具备深度学习基础，熟悉Diffusion、Transformer、DiT、VAE等生成模型结构与计算特点。
• 熟悉PyTorch推理链路，并至少精通TensorRT、ONNX Runtime、vLLM或SGLang之一。
• 理解CUDA与GPU内存模型，能够使用Nsight等工具做性能分析。
• 熟悉量化、算子融合、动态批处理、显存优化等加速方法。

优先条件（非硬门槛）

• 有分布式推理或模型并行经验。
• 有Stable Diffusion/SDXL或视频DiT/文生视频优化经验。
• 熟悉Diffusers源码、Triton/CUDA Kernel、国产芯片迁移或视频特有优化。