【工作职责】
我们是小红书中台大模型 Infra 团队,专注打造领先易用的「AI 大模型全链路基础设施」!团队深耕大模型「数-训-压-推-评」技术闭环,在大模型训练加速、模型压缩、推理优化、部署提效等方向积累了深厚的技术优势,基于 RedAccel 训练引擎、RedSlim 压缩工具、RedServing 推理部署引擎、DirectLLM 大模型 API 服务、QuickSilver 大模型生产部署平台等核心产品,持续赋能社区、商业、交易、安全、数平、研效等多个核心业务,实现 AI 技术高效落地!
【任职要求】
工作职责:
推理框架研发: 参与 rLLM 推理框架研发,支撑 LLM、MLLM、DiT 等模型的高性能在线推理。
分布式推理能力建设: 参与 KV Router、PD 分离 / EPD 分离、KV Cache 管理、Prefix Cache 复用、动态请求调度等核心能力建设。
推理性能优化: 围绕 TTFT、TPOT、吞吐、并发能力、GPU 利用率等指标,持续优化推理框架性能和资源效率。
模型压缩算法落地: 探索并落地大语言模型和多模态模型的低比特量化、蒸馏、剪枝、投机解码、KV Cache 压缩、CoT 压缩等技术,在保证业务效果的前提下降低推理成本。
压缩评估与部署闭环: 建设模型压缩评估与部署闭环,平衡精度、延迟、吞吐、显存、成本与业务效果,将压缩算法转化为稳定、可复用、可规模化的生产能力。
系统可靠性建设: 构建推理框架的系统容错能力,包括请求迁移、优雅退出、故障检测、自愈恢复、灰度发布等能力。
异构芯片推理适配: 负责推理引擎在国产异构计算芯片(昇腾 / PPU 等)上的适配与深度优化,完成模型迁移、算子开发、Profiling、Kernel 调优、并行策略与服务架构调优,推动国产算力在推理场景的规模化使用。
多模型与多硬件适配: 支撑 LLM、MLLM、Embedding、Rerank、DiT 等多类型模型在 Nvidia GPU、国产 NPU / PPU 等多种硬件上的推理、压缩和部署落地。
业务协同: 与算法、MaaS、平台和业务团队深度协作,为重点业务进行算法与系统联合优化,支撑业务推理成本下降和性能提升。