岗位职责
参与快手大模型推理引擎研发,工作内容包括:
- 参与大模型推理引擎的设计和研发,支撑快手自研以及开源模型的快速部署和高性能推理;
- 通过各种技术手段持续优化性能,降低推理成本,包括但不限于:算子/编译优化、异构推理、模型量化&蒸馏、分布式并行等;
- 支持RL中的多样化采样、generation性能优化等。
职位要求
- 有较强的学习能力和编程能力及数学基础;
- 有开源大模型推理框架(vllm/slang/trt-llm)使用和优化经验者优先;
- 有cuda或triton开发优化gpu算子经验者优先;
- 有昇腾、寒武纪、沐曦、海光、AMD 等算力芯片优化经验者优先;
- 有大模型推理相关从业经验者优先。